非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

  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值