import torchvision 失敗

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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章