分配了这个电脑,跑了10个程序9个因为cpu利用率太高,声音太大导致跑不了,让我一度以为是电脑散热坏了还有cpu不行,其实不是的
就是num_workers的设置问题,设置为1,cpu那个线程压力太大,所以设置为大点!!
可以设置为2,4这种,一般设置为4就行,设置太大,Gpu显存不够的话也处理不过来!
windows下设置num_workers会出现很多问题,一般的处理方式有两种,一种是将num_workers=0
另一种是将代码放在main()函数中
然后在最下面来一个就好
if __name__ == '__main__':
main()
还有一个问题
RuntimeError: CUDA error: an illegal memory access was encountered
这个错误解决
这样来。用了torch.cuda.set_device(0)后
把那些.to(device)都改成.cuda()就可以!!!