【PyTorch】PyTorch安装指南

PyTorch官网:http://pytorch.org/
由于官网各种点击按钮没有反应,只能下载安装。

【PyTorch】PyTorch安装指南_第1张图片

下载PyTorch

选择旧版本安装

【PyTorch】PyTorch安装指南_第2张图片

找到所需版本,如上图所示,下载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.post3-cp27-cp27m-manylinux1_x86_64.whl

或直接从链接安装

pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl

切记:在安装之前请先将已经安装的numpy卸载掉,安装PyTorch过程中会自动安装numpy,不然会报ImportError: numpy.core.multiarray failed to import错误。

安装torchvision

pip install torchvision

验证

import torch
import torchvision
print(torch.cuda.is_available())

若验证通过,则PyTorch已经安装成功。

参考:

http://blog.csdn.net/red_stone1/article/details/78727096

你可能感兴趣的:(PyTorch)