ubuntu上pip安裝pytorch1.1.0和torchvision0.4

官網找到相應的pytorch和torchvision版本並下載whl文件到本地,最新版conda安裝的話首頁直接複製命令就行:

其他版本需要進入以下鏈接:

https://download.pytorch.org/whl/cu100/torch_stable.html

選擇想要的pytorch和torchvision版本,cp36對應python3.6,複製鏈接地址下載。

進入文件保存的目錄,運行pip3 install torchvision-0.4.0-cp36-cp36m-manylinux1_x86_64.whl torch-1.1.0-cp36-cp36m-linux_x86_64.whl即可:

貌似torch-1.1.0對應的torchvision應該是0.3.0版本,torchvision0.4.0版本對應pytorch1.2及以上。但之前裝的torchvision-0.3.0跟cuda版本衝突,import報錯:ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory。(我安裝tensorfow的時候cuda裝的是10.0,當然找不到9.0)所以卸載重裝了torchvision0.4.0,目前沒啥問題,先就這樣了。

 

 

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