为了使用Visdom做可视化,所以得用cmd运行.py文件,在Pycharm中CUDA正常。在cmd中运行:
代码报错:
Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False.
但是我的CUDA是肯定能用的。百般排除无果后,在主函数第一句加上
print(torch.cuda.is_available())
尝试打印一下它是否可用,结果。。。。。。。。跑起来了
Visdom也运行正常了。。。。。。
无语。。。。。。。