ubuntu18.04離線安裝pytorch

      在線安裝一直無法建立連接,故採用離線的方法。

      首先確定本機的環境,包括默認的python版本,cuda版本。然後到:https://download.pytorch.org/whl/torch_stable.html下載所需的版本。如我的python版本爲3.6,cuda版本爲10.1,所以我下載了:cu101/torch-1.4.0-cp36-cp36m-linux_x86_64.whl,以及cu101/torchvision-0.5.0-cp36-cp36m-linux_x86_64.whl。其中,cu表示支持的cuda版本,cp表示支持的python版本。

      然後切換到相應目錄,分別安裝即可:    

sudo pip install torch-1.4.0-cp36-cp36m-linux_x86_64.whl
sudo pip install torchvision-0.5.0-cp36-cp36m-linux_x86_64.whl

       安裝完成後新建終端驗證 :

       python

       import torch

       import torchvision

       import torch.utils.data

       若沒有錯誤表明安裝成功。

       在此記錄安裝過程中的一些錯誤:

       1、使用pip安裝時出現“ModuleNotFoundError: No module named 'pip._internal'”

         解決方法:

         sudo vim /usr/local/bin/pip
         註釋from pip._internal.cli.main import main
         插入from pip import main

         安裝完成後不要忘了將上述操作還原。

       2、安裝.whl文件時出現“is not a supported wheel on this platform”

         解決方法:

         確保當前python版本是否與pytorch對應(切換本機python版本方法自行尋找,update-alternatives指令,另ubuntu18.04自帶python2.7與python3.6)。

       3、缺少各種庫,自行按提示安裝。

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