如何使用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
Uh oh!
There was an error while loading. Please reload this page.
-
看到requirement.txt，大概的想法是替换rocm版本的torch、支持rocm的transformers v4.36？其他代码中会有什么特化吗？
Beta Was this translation helpful? Give feedback.
All reactions