PyTorch安装

step1: 卸载numpy
首先要先卸载numpy,Torch过程中会自动安装numpy,不然会报ImportError: numpy.core.multiarray failed to import错误
step2: 安装PyTorch
PyTorch官网

pip install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp27-cp27mu-linux_x86_64.whl
pip install torchvision

step3: 安装Torchvision

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

若验证通过,输出True,则安装成功。

你可能感兴趣的:(Linux,python)