ubuntu16.04+tensorflow1.4+Cuda8.0+CUDNN7.0 系统安装

踩了不少的坑,花了将近一周才把东西弄出来,分享下踩坑经验,愿来者不再入坑。

参考资料_驱动以及cuda安装:
参考资料_cudnn以及tensorflow安装:http://blog.csdn.net/u014595019/article/details/53732015;

1. 系统环境:
ubuntu16.04 + cuda8.0 + cudnn6.0 + tensorflow1.4 + 1080Ti

cuda 9.0对应的cudnn是7.0,但是tensorflow必须是1.6或者1.5的
cuda 8.0对应的cudnn是6.0,tensorflow必须是1.4

否则会报错。

2. 驱动安装
对于1080Ti的英伟达驱动,需要的驱动程序版本是390,这边必须采用在线安装的方式最为靠谱。
安装需要依次执行以下四句:
sudo add-apt-repository ppa:graphics-drivers/ppasudo apt-get updatesudo apt-get install nvidia-390sudo apt-get install mesa-common-devsudo apt-get install freeglut3-dev
执行完毕之后重启即可.
sudo reboot
驱动检验,执行以函数:
nvidia-settings
或者:
nvidia-smi
如果有NVIDIA X Server Settings弹出,则安装成功

3.  安装Cuda
下载官方cuda8.0的runfile文件即可
在下载目录执行如下语句:
sudo sh cuda_8.0.27_linux.run
需要注意的是,不要安装驱动,驱动之前已经安装过了
安装完毕之后需要添加环境变量:
sudo gedit /etc/profile
打开“profile”文件,在末尾处添加(注意不要有空格,不然会报错):
export PATH=/usr/local/cuda-8.0/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH
重启电脑:
sudo reboot
测试是否安装成功:
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuerysudo make. /deviceQuery
如果出现关于GPU的信息,则表示安装成功。

4.  安装CuDNN
CUDNN的版本一定要对。tensorflow1.4对应的版本需要CUDNN6.0.
先去官网下载相应的cuDNN安装包,对其进行解压:
· tar -xvf cudnn-8.0-linux-x64-v5.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.hsudo chmod a+r /usr/local/cuda/lib64/libcudnn*
5.  安装tensorflow
首先要卸载原先的tensorflow,
sudo pip uninstall tensorflow
在线安装新的tensorflow:
pip3 install tensorflow-gpu==1.4
安装完成之后测试使用:
import tensorflow as tf
sess = tf.Session()
即可查看是否能够正常使用

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页