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列表,则驱动安装成功
Unbuntu 安装tensorflow-gup详解

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.****.net/ShellDawn/article/details/80274534
https://www.cnblogs.com/hutao722/p/9342577.html