ubuntu 16.04 安装TensorFlow-GPU

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zzz12122/article/details/82795732

@author: 向天笑
QQ:1500879657

1 安装前小贴士

贴士1:至今(2018.8.1)为止bantu 18.04对应打各种版本都试过,即使cuda安装成功,tensorflow也安装不成功,所以现阶段不建议安装18.04版本,而是安装16.04或14.04;

贴士2:目前(2018.8.1)tensorflow(tensorflow 1.9版本)最高支持到的CUDA版本是CUDA9.0,所以不要下载9.0以上版本。

贴士3:若你安装的是高版本的TensorFlow-GPU,不用降低自己的GCC到gcc-4.8版本,建议直接安装即可,若是gcc版本不对,则在安装nvidia driver环节会报错,那时再降低版本即可。

贴士4:在安装TensorFlow并允许时经常会出现:“ImportError: libcudnn.so.x(此处x表示版本号的一个整数): cannot open shared object file: No such file or directory”的错误。其原因是没有安装支持本机TensorFlow-GPU的cudnn或cudn所致,依据我的经验这种情况

方法一:换一个其它版本的TensorFlow-gpu(最方便,在pycharm上很容易实现)

方法二:重新安装其它版本的cuda或cudnn(比较麻烦)

方法三:设法安装一个和当前cuda或cudn版本可共存的其它cuda或cudn(至今未实现过2018.9.21)

2 安装显卡驱动

2.1查看自己打显卡:lspci |grep VGA

注意:在ubuntu上查看显卡适合的驱动理论上是可以的但是,我测试过上述命令,然并不容易看懂,在这里建议一下方法

2.1.1 方法一:首先查看自己电脑的牌子和型号,然后在京东、淘宝、中关村在线等平台上去查。

2.1.2 方法二:在Windows下查看自己电脑显卡驱动,再下载对应GPU驱动:可以看到更详细的信息。

2.2 在应nvidia观望查找适合电脑打显卡驱动并下载:驱动程序 | GeForce

3 安装cuda和cudnn

参考文章(本人安装时依第一个为主,第二个次之):

  (1)史上最全的ubuntu16.04安装nvidia驱动+cuda9.0+cuDnn7.0 - CSDN博客

(2)深度学习应用系列(一)| 在Ubuntu 18.04安装tensorflow 1.10 GPU版本

(3)真实机下 ubuntu 18.04 安装GPU +CUDA+cuDNN 以及其版本选择(亲测非常实用)

注意:安装cuda时加上命令--no-opengl-libs,不加这个选项会进入循环登陆。

例如:在终端运行指令 sudo sh cuda_8.0.27_linux.run --no-opengl-libs

4.安装与检测

参考文章:

CUDA安装和测试 - CSDN博客

猜你喜欢

转载自blog.csdn.net/zzz12122/article/details/82795732