非root用户在服务器（linux-Ubuntu16.04）上安装cuda和cudnn，tensorflow-gpu1.13.1

本文主要是介绍非root用户在服务器（linux-Ubuntu16.04）上安装cuda和cudnn，tensorflow-gpu1.13.1，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

1.准备工作（下载CUDA10.0和cudnn安装包）

查看tensorflow和CUDA，cudnn的版本的对应关系，从而选择合适的版本进行下载

下载CUDA10.0安装包，点击官网进行下载，根据服务器的具体情况选择对应的版本，如下图所示

下载完毕后得到安装包cuda_10.0.130_410.48_linux.run

下载cudnn，选择CUDA10.0对应的版本（需要注册登录nvidia账号），点击官网进行下载，选择“cuDNN Library for Linux”，下载得到cudnn-10.0-linux-x64-v7.6.5.32.solitairetheme8

将cudnn-10.0-linux-x64-v7.6.5.32.solitairetheme8重命名为cudnn-10.0-linux-x64-v7.6.5.32.tgz

建立好CUDA安装的自定义目录，例如/home/xxx/cuda-10.0

2.安装cuda

在安装包的目录下，输入命令进行安装

sh cuda_10.0.130_410.48_linux.run

按q退出协议阅读，输入accept接受协议，输入选项进行安装，如下图

等待安装即可。

3.安装cudnn

首先对cudnn-10.0-linux-x64-v7.6.5.32.tgz进行解压，命令为

tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.tgz

解压后的文件名为cuda，然后进行下面的操作

cp /home/xxx/cuda/include/cudnn.h  /home/xxx/cuda-10.0/include/  
cp /home/xxx/cuda/lib64/libcudnn*  /home/xxx/cuda-10.0/lib64
chmod a+r /home/xxx/cuda-10.0/include/cudnn.h  /home/xxx/cuda-10.0/lib64/libcudnn*

4.修改环境变量

需要修改.bashrc中的环境变量，在文件最后面添加

export PATH=/home/xxx/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/home/xxx/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/home/xxx/cuda-10.0/

然后刷新文件，使环境变量生效

source .bashrc

5.检查CUDA的版本和cudnn的版本

输入下面两个命令都可以检查CUDA10.0的版本

nvcc -V

 

cat /home/xxx/cuda-10.0/version.txt

输入命令检查cudnn的版本

cat /home/xxx/cuda-10.0/include/cudnn.h | grep CUDNN_MAJOR -A 2

6.安装tensorflow-gpu

使用anaconda进行安装tensorflow，如果没有下载可以点击官网先下载再安装

下载后输入命令进行安装

bash Anaconda3-5.2.0-Linux-x86_64.sh

创建一个tensorflow-gpu环境并指定python版本，命令如下

conda create -n tensorflow-gpu python==3.6

创建完成后激活环境

source activate tensorflow-gpu

输入命令安装tensorflow-gpu

pip install tensorflow-gpu==1.13.1

7.验证tensorflow-gpu是否安装成功

在tensorflow-gpu的环境下，输入python进入终端，然后输入下面的命令

import tensorflow as tf
print('GPU',tf.test.is_gpu_available())

显示“GPU True”,即代表GPU版本安装成功！

参考文献

https://blog.csdn.net/qq_45032341/article/details/101112756

https://blog.csdn.net/Mr_WangYC/article/details/104234733

https://blog.csdn.net/gyx1549624673/article/details/104502561/

https://blog.csdn.net/dlh_sycamore/article/details/107600717

https://blog.csdn.net/baidu_32936911/article/details/79774289

这篇关于非root用户在服务器（linux-Ubuntu16.04）上安装cuda和cudnn，tensorflow-gpu1.13.1的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---