1 官方推薦:
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
2 官方不建議使用的
torch.cuda.set_device(1) # 設定GPU ID
3 在終端中指定
CUDA_VISIBLE_DEVICES=1 python my_script.py
1 官方推薦:
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
2 官方不建議使用的
torch.cuda.set_device(1) # 設定GPU ID
3 在終端中指定
CUDA_VISIBLE_DEVICES=1 python my_script.py