添加一个选择显卡的功能 #277
Closed
fengweipeng
started this conversation in
Ideas
添加一个选择显卡的功能
#277
Replies: 1 comment
-
请教秋叶大神,如果是需要两张显卡一起训练,丹炉里应该怎么设置呢?我选择了多显卡选项,但是训练直接报错。 |
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
-
就像“绘世”中的那样,如果,电脑上有双显卡,可以选择要用那一张显卡。
现在,程序会使用系统默认的显卡进行训练。如果有双显卡,需要选择第二张显卡进行训练,就要手动修改:A启动脚本.bat
在
set PYTHON=python\python.exe
后面添加一行:
set CUDA_VISIBLE_DEVICES=1
变成:
Beta Was this translation helpful? Give feedback.
All reactions