【PyTorch】PyTorch安装指南
·
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.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已经安装成功。
参考:
更多推荐



所有评论(0)