天天看點

pytorch 指定 GPU

1 官方推薦:

os.environ["CUDA_VISIBLE_DEVICES"] = "1"
           

2 官方不建議使用的

torch.cuda.set_device(1)  # 設定GPU ID
           

3 在終端中指定

CUDA_VISIBLE_DEVICES=1 python my_script.py