Pycharm远程运行指定GPU
作者:快盘下载 人气:目录
1.源代码中添加
2.在gpu中运行时添加
3. 不添加源代码
1.源代码中添加
import os
# 指定使用0,1,2三块卡
os.environ[;CUDA_VISIBLE_DEVICES;] = ;0,1,2; 2.在gpu中运行时添加
CUDA_VISIBLE_DEVICES=0,1,23. 不添加源代码
加载全部内容
目录
1.源代码中添加
2.在gpu中运行时添加
3. 不添加源代码
import os
# 指定使用0,1,2三块卡
os.environ[;CUDA_VISIBLE_DEVICES;] = ;0,1,2; CUDA_VISIBLE_DEVICES=0,1,2
加载全部内容