pytorch入门之安装和配置

pytorch是一种python接口的深度学习框架，其他的框架还有caffe，tensorflow等等。

1，pytorch目前支持linux和OSX两种系统。支持的Python版本有2.7，3.5，3.6。

2，包管理工具PackageManager我们选择conda，在Anaconda这个功能强大的包中包含了conda，也包含python以及很多python的扩展工具包。在需要的时候，选择下载安装就可以了。

3，学习深度学习最好有支持cuda的NVIDIA显卡，用来进行快速的计算。

在这里我们需要做的是：

一，安装显卡驱动，在Ubuntu上，在SystemSettings--》Software&Updates--》AdditionDrivers中联网状态下，会自动搜索可以使用的显卡，点击ApplyChanges，重启电脑，显卡驱动生效。

二，安装CUDATookit，pytorch最多支持到CUDA8.0，要在官网找到相应的下载包括了cuda8.0和对应的cudnn的安装。

安装完以后要进行相关的配置，在home目录中的.bashrc文件中，在末尾添加一下代码：

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
          export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
          export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

4， 到现在才是真正的安装pytorch

安装pytorch需要安装两个东西，pytorch和torchvision。用相关命令安装即可

5，验证安装是否成功

sudo lshw -c video 查看显卡信息

nvcc -V  显示CUDA信息

python进入python环境，import  torch   import  torchvision 不报错就是成功了