摘要:
实验室的同学一直都是在服务器上既用CPU训练神经网络也有使用GPU的,最近才发现原来在pytorch中可以通过设置 torch.set_num_threads(args.thread) 来限制CPU上进行深度学习训练的线程数。 torch.set_num_threads(args.thread) 在 阅读全文
posted @ 2022-01-29 09:12
Angry_Panda
阅读(11725)
评论(0)
推荐(0)