1
OMP_NUM_THREADS=1
export OMP_NUM_THREADS
在服務器上跑PyTorch 程序的列位不要把cpu佔滿
修改這個環境變量降低你們的cpu使用率
(因爲pytorch默認是能用多少線程就用多少的)
在python代碼中:
# 選擇顯卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# 只使用一個線程
os.environ["OMP_NUM_THREADS"] = "1"
2
這種方法經常失效,建議使用1中的方法
pytorch中設置線程個數
torch.set_num_threads(int)
3
選擇用哪個顯卡來跑程序
CUDA_VISIBLE_DEVICES=1 python xx.py