ubuntu18.04装tensorflow遇到的问题

2018年5月21日更新
关于英伟达驱动的卸载与安装:点击打开链接

这个帖子非常有用(cuda,cudnn,tensorflow的安装):https://www.tinymind.cn/articles/163

根据帖子里所说安装了cuda9.1+cudnn7.1+tensorflow非官方版(官方版到今天只支持cuda9.0),由于之前安装nvidia驱动时候把系统装崩了,所以直接从头装系统了,然后第一步就先装驱动,笔者用的GEFORCE 1080 Ti,所以下载了当时最新的驱动390.48,根据网上说的方法安装,如添加blacklist,关闭X等。笔者中间遇到没有gcc的问题(由于重新装了系统没有安),并且使用sudo apt-get install gcc会出错,找了好半天没有用,后来尝试sudo apt-get install g++,然后再sudo apt-get install gcc就可以了,之后就按着上面这个帖子所说,直接使用sudo aptitude install nvidia-cuda-toolkit就可以安装cuda9.1(应该是默认使用这个命令的最新版,不过笔者安装时已经有9.2,但是该命令装的是9.1,请大家注意提示,如果提示安装是9.2请三思处理),后面就顺风顺水,tensorflow使用作者给的github链接下载即可。


----------------------------------------------------------------------------------------------------------

sudo apt install python3-pip

pip install tensorflow==1.8

显示安装成功,在python里尝试import:

猜可能是因为18.04太新的原因,导致不支持。

网上查阅说可以直接装tensorflow gpu:

(如果不用gpu可以参考https://github.com/tensorflow/tensorflow/issues/8037解决问题)

pip install tensorflow-gpu==1.8.0

然后就连tensorflow都无法import了。


又经过一番查阅折腾,发现安装gpu版本前需要先装CUDA和CuDNN驱动,我就又把tensorflow卸载了,打算重装,因为同时发现应该使用pip3来安装python3.n版本的(原谅小白),卸载:

sudo pip uninstall protobuf
sudo pip uninstall tensorflow

然后下载CUDA的时候发现踩坑,只支持到ubutnu17.04(我用的18.04). 打出GG

又在nvidia论坛上看到有人说18.04也支持,于是又开始尝试。

尝试在nvidia官网上下载cuda9.1 for ubuntu17.04,发现cuda被墙了:

哇,真的头疼。。。

先用cpu版学习一下再说



阅读更多
文章标签: tensorflow
下一篇ubuntu18.04 配置python tensorflow虚拟环境+pycharm
想对作者说点什么? 我来说一句

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

关闭
关闭
关闭