PyTorch 安装

安装PyTorch

　　PyTorch 官网 http://pytorch.org/

　　在选好环境后会有相应的安装提示：

　　　　我的环境是Ubuntu系统，选用pip安装，Python3.5, CUDA 版本为8。

　　　　在配置过程中，CUDA的版本有些模糊，使用命令 

　　

cat /usr/local/cuda/version.txt

 

　　　　结果为

CUDA Version 8.0.61
CUDA Patch Version 8.0.61.2

　

安装命令

　

pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.1-cp35-cp35m-linux_x86_64.whl 
pip3 install torchvision

　　下载安装包大概一个多小时，在此期间，可以到官网的教程 学习一下 http://pytorch.org/tutorials/ 和 官方例程 https://github.com/pytorch/examples