一、前言
安装下最新的Tensorflow2(GPU版本),从头到尾记录下步骤。
从显卡驱动到CUDA、cuDNN,再到tensorflow-gpu。
记录于2019年11月22日
二、环境配置
UBuntu 18.04
2.1 预定安装版本
nvidia-driver-440
CUDA 10.0+cuDNN 7.6.4 + tensorflow-gpu-2.0.0 + python3.7
注:根据自己电脑配置和需求自己找适合自己的版本即可,从driver版本开始选择。
具体对应版本号可看下面“三、配置对应版本号”
2.2 如何选择自己电脑的安装版本
想要自己更改版本下载自己想要版本的的可以往下看,
想要简单点直接用的话,按照这个版本:
nvidia-driver-440+CUDA 10.0+cuDNN 7.6.4 + tensorflow-gpu-2.0.0 + python3.7的话直接跳到第四步安装看即可。
1、从tensorflow版本 入手,选择合适的CUDA、cuDNN、nvidia-driver版本。
这里我以tensorflow-gpu-2.0.0为例子。
2、查看下方的"三、配置对应版本号"的第一个版本配置表格我们可以看出。
tensorflow-gpu-2.0版本 对应如下:
这里我选择 CUDA 10.0, cuDNN7.6.4(官网支持CUDA10.0的cuDNN最新版本)。
3、查看自身的Python版本、GCC版本是否满足需求
查看已安装Python版本
python -V
# Python 3.7.4
查看已安装GCC版本
gcc -v
# gcc version 7.4.0 (Ubuntu 7.4.0-1ubuntu1~18.04.1)
Python、GCC接满足需求。
4、选择 nvidia-driver版本