想了一下还是把tensorflow安装的过程整理一下吧,万一时间久了忘了呢。
1、准备工作
1、更换源,好的软件源,直接决定你的安装速度。这里选择清华的。
操作:进入:设置 --> 软件和更新 --> Ubuntu软件,打开下载自:选择 其他,选择清华。点击 选择服务器 确认,在弹出框选择,重新载入。
2、更新软件包,防止有些工具没有安装上。
sudo apt update
3、打开:设置 --> 软件和更新 --> 附加驱动
选择NVIDIA公司专有的最新驱动,选择应用更改。时间较长,请耐心等待。
2、安装cuda
下载文件,这里我选择run本地文件,进入文件所在目录,并运行,XXX代指文件名
sudo sh XXX.run
安装注意,这时会出现很多信息,按ctr+c关闭信息,再按accept接受。
接下来询问是否,安装驱动,输入n,不要安装。因为已经安装过。
一路输入y,或者回车。
进入/home/用户名/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery,打开终端,输入
sudo make
然后输入
./deviceQuery
如果显示pass,说明cuda安装成功,如果显示说,cuda驱动不满足于cuda运行库,说明驱动太旧,或者cuda太新。参照修改。
3、配置cudnn软链接
cd /home/***(自己的用户名)/Desktop/###(这个命令意思是找到刚刚我们用U盘传过来的文件) tar xvzf cudnn-8.0-linux-x64-v5.1-ga.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.h /usr/local/cuda/lib64/libcudnn*
4、安装tensorflow
在配置完cudnn之后,可以直接在终端输入
sudo pip install tensorflow==1.5.0
或者安装CPU版本
sudo pip install tensorflow==1.5.0
如果网速不好的话,也可以自己在https://github.com/tensorflow/tensorflow下载whl文件,往下翻,在installation下面,然后使用
sudo pip install xxx.whl