Anaconda CUDA9.1.83對應的pytorch安裝,不同版本的pytorch下載


CUDA版本是 9.1.83,python版本3.6.5 win10

使用如下語句可以安裝cu91版本的pytorch
pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
然而下載torchvision會使用之前緩存中的torchvision而且0.4.1版本的並不能使用

最后使用

pip install torchvision==0.1.8

成功安裝

選擇對應的版本:

進入網頁選擇合適的pytorch版本:

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

image

其中CUDA后面的數字對應自己電腦上CUDA的型號。
其中cp35表示環境python匹配的版本
當然根據電腦系統的不同,也有着不同系統版本的選擇。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM