pytorch安裝


服務器是的cuda9,python3.6,unbuntu16

pytorch之前的版本:http://pytorch.org/get-started/previous-versions/

 

之前下載的cu90/torch-0.4.0-cp36-cp36m-linux_x86_64.whl,用pip3 install 這個whl會報錯:

torch-0.4.0-cp35-cp35m-linux_x86_64.whl is not a supported wheel on this platform.

發生的

這個是pytorch github的圖,可以看到python3.6只支持ppc64le,https://zhidao.baidu.com/question/4887034.html這個上面說ppc64le是mac的

我就換成了cu90/torch-0.4.0-cp27-cp27m-linux_x86_64.whl,同樣是錯誤,后來看了一下網上的一般是用cu90/torch-0.4.0-cp27-cp27mu-linux_x86_64.whl

最后換成了cu90/torch-0.4.0-cp35-cp35m-linux_x86_64.whl,用pip install安裝也抱同樣的錯誤,用pip3 install就能正確安裝了

安裝成功后再安裝sudo pip install torchvision

最后python3.5,import torch就可以看到安裝成功,但是python2,python3.6都會報錯

 

總結:去下載對應版本的whl,然后pip install whl,然后pip install torchvision


免責聲明!

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



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