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,2

3. 不添加源代码