在线安装一直无法建立连接,故采用离线的方法。
首先确定本机的环境,包括默认的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、缺少各种库,自行按提示安装。