安装pythorch

下贼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'
>>> 

 

你可能感兴趣的:(配置环境)