pytorch指定gpu以及kill掉某個進程

1. 在運行代碼的時候在終端設置 CUDA_VISIBLE_DEVICES=0 python train.py 2.在代碼中指定 import os os.environ[「CUDA_VISIBLE_DEVICES」] = 「0」 3. set_device import torch torch.cuda.set_device(0) 以上方式意味着指定使用0號GPU，官方建議使用CUDA_VISI

>>阅读原文<<