几个常用线程的

  1. 云栖社区>
  2. 博客>
  3. 正文

几个常用线程的

毛球饲养员 2017-10-08 10:56:26 浏览2727
展开阅读全文
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

网友评论

登录后评论
0/500
评论