进入PyTorch官网:
PyTorch
在PyTorch主页上右下角点击“Click here for previous versions of PyTorch”,如下图:
然后在弹出的界面,根据你自身装的CUDA版本去选择对应的whl文件进行下载:
要用GPU的话,就要选择含cu的版本,我的电脑安装的是CUDA 8,然后我想安装在系统的Python3.5上,所以选择如下:
其中cp35代表python3.5,同理,cp27就是python2.7
前面的0.4.0或者0.3.1就是PyTorch的版本数,建议使用0.3.1,因为现在很多开源的代码都是基于0.3的,不同版本的函数有些不一样
进入PyTorch的下载目录,使用pip3命令安装:(python3是用pip3,python2就直接用pip,根据自己的情况,注意这点不同)
sudo pip3 install torch-0.3