平时编程常用的

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

你可能感兴趣的:(平时编程常用的)