tensorflow-gpu1.8安装

一、安装GPU显卡驱动

显卡驱动是安装tensorflow最操蛋的,也是最难受的,深受其害,耽误半天时间。

按装这个驱动之前首先要确定你电脑的显卡是英伟达的,否则累死你也做不上,还骂了半天的娘。

网上最常用的方法如下:

ubuntu16.04中,更换驱动非常方便,去

系统设置->软件更新->附加驱动->切换到最新的NVIDIA驱动即可。应用更改->重启


然后输入nvidia-smi命令


如果没有这个框的话很有可能,就是nvida的版本太低,你要重新下载一个新版本,越新越好。亲测有效,解决了大毛病。

更新nvidia方法:

sudoapt purge nvidia-*

sudoadd-apt-repository ppa:graphics-drivers/ppa

sudoapt update

sudoapt install nvidia-387

二、安装JDK8

sudoapt-get install software-properties-common

sudoadd-apt-repository ppa:webupd8team/java

sudoapt-get update

sudoapt-get install oracle-java8-installer

验证java版本:java-version

三、安装CUDA9.0

安装cuda时要清楚你的软件支持哪一版的。

下载地址:https://developer.nvidia.com/cuda-downloads



四、安装CUDNN7.1.4

下载地址: https://developer.nvidia.com/cudnn



tar -xvf cudnn-9.0-linux-x64-v7.1.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

五、配置环境变量

sudo gedit ~/.bashrc

exportLD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64”

export CUDA_HOME=/usr/local/cuda

exportPATH="$CUDA_HOME/bin:$PATH"



source ~/.bashrc

六、安装TensorFlow

1、安装依赖工具

sudoapt-get install python-numpy swig python-dev python-wheel

2、用pip安装很方便

pipinstall tensorflow_gpu==1.8

问题:

1、没有libcudnn.so.7cd

cd /usr/local/cuda/lib64/

sudo rm -rf libcudnn.so libcudnn.so.7

sudo ln -s libcudnn.so.7.1.4 libcudnn.so.7

sudo ln -s libcudnn.so.7 libcudnn.so




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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值