几个常用线程的

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