PyTorch官网:http://pytorch.org/
由于官网各种点击按钮没有反应,只能下载安装。
下载PyTorch
选择旧版本安装
找到所需版本,如上图所示,下载cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl。其中,0.2.0表示PyTorch版本,cp27表示支持Python2.7,cu80表示支持CUDA 8.0。
安装PyTorch
进入PyTorch的下载目录,使用pip命令安装:
pip install torch-0.2.0.p