根据帖子里所说安装了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版学习一下再说