ubuntu18.04 安装NVIDIA显卡驱动+CUDA+CUdnn+tensorflow-gpu1.13+pytorch

写下此文档方便下次安装。其实安装别的NVIDIA驱动也都是这个流程

1  首先安装系统，这个使用universal USB installer 小工具制作U盘启动。很方便

2 安装好系统，把nouveau 禁用掉

3 去https://www.nvidia.cn/geforce/drivers/  把NVIDIA 驱动装好

3 接下来就是装并行计算框架CUDA和神经网络加速库cudNN

  CUDA链接：https://developer.nvidia.com/cuda-toolkit-archive  

  配置环境：(目的是让tensorflow,pytorch能找到cuda)

   

#以下命令我是用root用户编辑的
vim ~/.bashrc
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64

PATH=$PATH:/usr/local/cuda-10.1/bin

CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

cudnn 下载链接（要和CUDA版本对应一致）：https://developer.nvidia.com/rdp/cudnn-archive

执行以下命令：

#如果安装过cuda 并创建过链接
cp cuda/include/* /usr/local/cuda-10.1/include/

cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64/

chmod a+r /usr/local/cuda-10.1/include/cudnn.h

chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

查看cudnn版本信息：

cat /usr/local/cuda-10.1/include/cudnn.h | grep CUDNN_MAJOR -A 2

出现版本信息则安装成功

#define CUDNN_MAJOR 7

#define CUDNN_MINOR 6

#define CUDNN_PATCHLEVEL 5

安装tensorflow-gpu1.13.1

  

#ubuntu18.04 自带python3.6 所以需要安装pip
apt install software-properties-common
apt-get install python3-pip

pip3 install -i  https://pypi.doubanio.com/simple/ tensorflow-gpu==1.13.1

安装pytorch

 pip3 install numpy==1.12.1 -i https://pypi.doubanio.com/simple/

 pip3 install torch===1.3.1 torchvision==1.3.1 -i  https://pypi.doubanio.com/simple/