在任务管理器和设备管理器查看自己的gpu配置
在代码中设置使用gpu运行代码
import os
# 使用第一张与第三张GPU卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0, 2"
具体安装和配置可参考:
在任务管理器和设备管理器查看自己的gpu配置
在代码中设置使用gpu运行代码
import os
# 使用第一张与第三张GPU卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0, 2"
具体安装和配置可参考: