欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-03-27 09:44:04
...

一、安装顺序

  • 按照显卡驱动、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)

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

安装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

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

相关标签: 环境搭建