torch.set_num_threads(all_args.n_training_threads) 可降低CPU使用率

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

 

torch.set_num_threads(all_args.n_training_threads) 

posted @ 2021-11-26 11:49  呦呦南山  阅读(565)  评论(0编辑  收藏  举报