平时编程常用的

linux占用一个线程

OMP_NUM_THREADS=1
export OMP_NUM_THREADS
在服务器上跑PyTorch 程序的列位不要把cpu占满
修改这个环境变量降低你们的cpu使用率
（因为pytorch默认是能用多少线程就用多少的）

pytorch中设置线程个数

torch.set_num_threads(int)

选择用哪个显卡来跑程序

CUDA_VISIBLE_DEVICES=1 python xx.py