幾個常用線程的

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