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