import torchvision 失败

时间:2025-03-18 16:30:11

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

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

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

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

conda uninstall torchvision

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

/get-started/locally/ 打开官网,选择你相应的配置

conda的一些集合:

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