Ubuntu16.04 安装Cuda Cudann 配置GPU版本pytorch环境

1.安装NVIDIA的驱动:点击win键，search your computer=>搜索 Additional Drivers =>选择第一个 =>restart

2.Cuda8.0版本：https://developer.nvidia.com/cuda-80-ga2-download-archive

下载完版本之后进行安装包目录：

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

3.Cudann8.0版本：解压 cudnn-8.0-linux-x64-v6.0.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include # 复制到 include 中
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 # 复制到 lib64 中
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* # 将头文件复制进去

4.ubuntu的gcc编译器是5.4.0，然而cuda8.0不支持5.0以上的编译器，因此需要降级，把编译器版本降到4.9：

sudo apt-get install gcc -4.9 gcc-5 g++-4.9 g++-5

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10

sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20

sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10

sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30

sudo update-alternatives --set cc /usr/bin/gcc

sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30

sudo update-alternatives --set c++ /usr/bin/g++

5.配置cuda8.0之后主要加上的一个环境变量声明，在文件~/.bashrc之后加上(note: /usr/bin/sudo gedit ~/.bashrc)

gedit ~/.bashrc

export PATH="/usr/local/cuda-8.0/bin:$PATH"

export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:$PATH"

6.然后设置环境变量和动态链接库:

sudo gedit /etc/profile

在打开的文件里面加上:（注意等号两边不能有空格）

export PATH=/usr/local/cuda/bin:$PATH

保存之后，创建链接文件:

sudo gedit /etc/ld.so.conf.d/cuda.conf

在打开的文件中添加如下语句：

/usr/local/cuda/lib64

保存退出执行命令行,使链接立即生效:

sudo ldconfig

6.执行以下命令测试cuda是否安装成功:

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery

7.pytorch:https://pytorch.org/get-started/locally/

conda install pytorch torchvision cuda80 -c pytorch

https://blog.csdn.net/QLULIBIN/article/details/80728355

https://www.cnblogs.com/xujianqing/p/6142963.html

https://blog.csdn.net/iotlpf/article/details/54175064