在服务器上跑CNN网络代码时出错。
先查看python环境中torch和cuda版本。
import torch
print(torch.__version__)
print(torch.version.cuda)
结果显示
torch 1.8.0 cuda 10.2
看网上说可能是torch和cuda版本不匹配。于是我删除conda虚拟环境中的pythoch环境。
conda remove pytorch
接着重新安装torch
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
安装后发现问题并没解决。
后来到pytorch官方网站重新安装最新版。https://pytorch.org/
问题解决了,深度学习程序成功在服务器上运行成功,并跑出来了预期结果。