当你的主机上有多张GPU卡,可以通过如下方式设置代码main.py使用哪几张GPU卡来运行程序
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0,1,2" # 设定使用0,1,2号GPU卡
然后执行命令python main.py
即可
注意,上述两行代码要作为main.py的第1、2行
当你的主机上有多张GPU卡,可以通过如下方式设置代码main.py使用哪几张GPU卡来运行程序
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0,1,2" # 设定使用0,1,2号GPU卡
然后执行命令python main.py
即可
注意,上述两行代码要作为main.py的第1、2行