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