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
