pytorch安装

pytorch安装

在http://pytorch.org/中点击对应的系统版本,即可以在网页上的run this command找到对应的命令行安装代码.

我的电脑ubuntu 同时安装了python2.7和python3.5,安装pytorch for python3.5时,按照pytorch的安装教程终端输入命令行:sudo pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl

报错:

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

这是查看pyhon版本python –version,发现pyhon代码为python 2.7

由于我是要安装pytorch到python3,因此将python切换到python3,命令如下:

$alias python=’/usr/bin/python3’$ python –version
Python 3.5.2 # 版本已经切换到python3

重新输入安装命令,还是报同样的错,这是查看pip版本,命令为:

$pip –version

$pip 9.0.1 from /usr/local/lib/python2.7/dist-packages (python 2.7)

也就是说当前的pip is for python2.7,对于python3其对应的pip为pip3,

因此将安装命令中pip改为pip3,即

$sudo pip3 install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl

安装成功

在命令行输入:$python

输入import pytorch不出错,说明安装pytorc成功

另外,安装python3的库时,使用sudo apt-get install python3-file 例如 安装nltk,sudo apt-get install python3-nltk,或者使用pip3安装即可