pytorch 版本问题
pytorch官网:https://pytorch.org/get-started/locally/
服务器上的cuda 10.0已经不受最新版本支持,需要去历史版本找。
找到的官方源太慢,可以更换为清华源
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
到这里确认系统:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
查看cuda版本
nvcc -V