ubuntu16.04 + cuda 9.0 +cudnn7.0 +(gcc-4.8,g++ -4.8) +tensorflow-gpu1.6.0(1.5.0)

一、安装顺序

  • 按照显卡驱动、cuda、cudnn、gcc,g++、tensorflow 的顺序安装;
  • 最重要的是版本的匹配问题;
  • 每次安装完都要进行测试是否安装成功,确保之前的工作是有效的;
  • 环境比那辆要配置正确,所有东西安装完毕之后重启测试。

二、检查本显卡支持的驱动的最高版本

软件和更新里面查找 附加驱动
ubuntu16.04 + cuda 9.0 +cudnn7.0 +(gcc-4.8,g++ -4.8) +tensorflow-gpu1.6.0(1.5.0)_第1张图片

找到支持显卡的最高版本的驱动,一定以自己的机器为准,官网上显示支持的版本可能比上面查找的要高,后面会出问题。
而后下载显卡驱动,安装显卡,参照网上教程。成功后重启验证。
(设置-详细信息)发现已经是自己安装的显卡则说明正确。
ubuntu16.04 + cuda 9.0 +cudnn7.0 +(gcc-4.8,g++ -4.8) +tensorflow-gpu1.6.0(1.5.0)_第2张图片

安装cuda

官网下载cuda,而后 sh 安装。
安装中 Ctrl +c 跳过文档,问候在选择是否安装驱动选择否,其余默认。
之后配置环境变量。
sudo gedit ~/.bashrc
加上:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存退出

source ~/.bashrc

在profile中也输入上面两行声明

sudo gedit /etc/profile

安装cuDNN

首先检查gcc和g++版本,Ubuntu自带的版本对于tensorflow太高,所以要降低到4.8.
下载对应cuda 的Cudnn
而后将需要的文件cp到相应目录,并且要生成软链接,删除原来的链接。

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

然后更新网络连接:

cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.7.0.5  //自己查看.so的版本 对应更改
sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig 

安装tensorflow-GPU

这个方法比较多,只要版本匹配就都可以。
补充,待更。。。

你可能感兴趣的:(环境搭建,TensorFlow)