下贼pytorch:http://pytorch.org/
要安装和你python版本还有cudn版本对应的包。
安装pytorch
进入PyTorch的下载目录,使用pip命令安装:
比如我下载的是0.4.0版本的
pip install torch-0.4.0-cp35-cp35m-linux_x86_64.whl
安装torchvision
安装torchvision比较简单,可直接使用pip命令安装:
pip install torchvision
python
>>> import torch
>>> torch.__version__
'0.4.0'
>>>