torch已經安裝成功了,import torch也沒有問題
torchvision已經安裝成功,如果沒安裝成功也可往下看
但是import失敗 問題 DLL:找不到模塊
原因應該是對應的torchvision的版本不匹配,先把現有的torchvision卸載
conda uninstall torchvision
再根據對應的cuda版本進行安裝,也可更具以下網址進行下載
https://pytorch.org/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