Unbuntu 安装tensorflow-gup详解

1.查看NVIDIA显卡驱动版本与cuda、cudnn的版本对应关系，有时候显卡驱动版本太低不支持，找到适合自己的驱动版本。

NVIDA驱动：https://www.nvidia.com/Download/index.aspx?lang=en-us

2.安装1中找到的显卡驱动版本

【注】试了很多种方法，这种是最简单的
通过命令行安装驱动：
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt-get update
$ sudo apt-get install nvidia-390 #此处数字要对应上面查询到的版本号

重启
$ sudo reboot

在终端验证是否安装成功：
$ nvidia-smi #如果出现GPU列表，则驱动安装成功

3.查看tensorflow-gpu版本和cuda、cudnn版本的对应关系，下载需要的版本

cuda官网：https://developer.nvidia.com/cuda-toolkit-archive
cudnn官网：https://developer.nvidia.com/rdp/cudnn-archive

4.安装cuda和cudnn

网上很多教程，这里就不赘述了
[注]需要安装gcc和g++，同时也要注意版本，可以通过增加或者降低版本的适应。

5.安装tensorflow-gpu

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==需要的版本

6.针对import tensorflow出现的错误解决

有时候会出现runtime错误
解决办法：pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade --ignore-installed tensorflow

7.其中一些细节可以参考的博客

https://blog.csdn.net/ShellDawn/article/details/80274534
https://www.cnblogs.com/hutao722/p/9342577.html