在線安裝一直無法建立連接,故採用離線的方法。
首先確定本機的環境,包括默認的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、缺少各種庫,自行按提示安裝。