Ubuntu18.04 + 1808Ti + CUDA 9.0 + CUDNN 7.0 + tensorflow-gpu

tensorflow-gpu安装了很多遍,给大家推荐一个没有坑的Ubuntu tensorflow-gpu 安装方法,亲测无坑:

主要参考文章:https://blog.csdn.net/u014670893/article/details/82467384

更改驱动:

将ubuntu驱动更改为NVIDIA版本驱动

安装完成后最好重启下。

检查驱动是否安装成功:

~$:nvidia-smi

 

安装CUDA Toolkit 9.0

 

下载地址:

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=runfilelocal

直接按照下图版本下载17.04的版本,18.04可以使用

    

下载完成后,终端输入以下代码安装:

     ~$:sudo chmod +x cuda_9.0.176_384.81_linux.run
     ~$:./cuda_9.0.176_384.81_linux.run --override

根据提示安装即可,    

installing with an unsupported configuration?时选择yes

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?为了稳定最好选择no

样例可以不安装。


安装CUDNN 7.0

地址:https://developer.nvidia.com/rdp/cudnn-archive

CD到下载的文件夹中

    # 解压
    ~$:tar -zxvf cudnn-9.0-linux-x64-v7.tgz
     
    # 复制相应文件
    ~$:sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/
    ~$:sudo cp  cuda/include/cudnn.h /usr/local/cuda-9.0/include/
     
    # 所有用户可读
    ~$:sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

libcupti

sudo apt-get install libcupti-dev

配置

编辑~/.bashrc

    ~$:gedit ~/.bashrc

在~/.bashrc中加入

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

安装tensorflow-gpu

也可以使用pip3

~$:pip install tensorflow-gpu==1.14.0 

网络不好可以多试几次,安装成功后,进入测试:

~$:python3

import tensorflow as tf

没有报错,则安装成功。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

VictorHan01

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值