如何使用AMD显卡进行微调？
#485
Replies: 2 comments
-
|
代码没有显式要求是cuda，可以试试 |
Beta Was this translation helpful? Give feedback.
0 replies
-
|
感谢，已经可用。安装rocm专用torch和bitsandbytes即可。 |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
看到requirement.txt，大概的想法是替换rocm版本的torch、支持rocm的transformers v4.36？其他代码中会有什么特化吗？
Beta Was this translation helpful? Give feedback.
All reactions