大概原因就是,装的pytorch版本太老或者没装好,得利用官网的命令重新装一次
进入
https://pytorch.org/get-started/previous-versions/
找到conda安装下的你的系统(windows)和CUDA版本对应的命令,复制下来。把后面的 -c pytorch 去掉
不过,在运行这个命令前,你最好先给conda添加国内镜像源。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
最关键的是最下面这一句,不过上面几句可以方便你以后安装包。
然后打开cmd窗口运行你复制下来的命令就好了
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2
附:如何看你的CUDA版本:
查看win10 CUDA版本
觉得对你有帮助就点个赞,收藏一下!
参考文献:
https://www.jb51.net/article/180523.htm