import torchvision 失败

torch已经安装成功了,import torch也没有问题

torchvision已经安装成功,如果没安装成功也可往下看

但是import失败 问题 DLL:找不到模块

原因应该是对应的torchvision的版本不匹配,先把现有的torchvision卸载

1
conda uninstall torchvision

再根据对应的cuda版本进行安装,也可更具以下网址进行下载

https://pytorch.org/get-started/locally/ 打开官网,选择你相应的配置

conda的一些集合:

1
2
3
4
5
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

cuda10.0:conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
cuda 9.0:conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
none cuda:conda install pytorch-cpu torchvision-cpu -c pytorch