【Tensorflow】Tensorflow 2.2.0 安装教程

起序:现在 Tensorflow 已经到 2.2.0 了,但是 CUDACUDNN 不能安装最新版的,因为痛的领悟。

会报错:Could not load dynamic library ‘cudart64_101.dll’; dlerror: cudart64_101.dll not found.

所以,先查询好 Tensorflow 和 CUDA 及 CUDNN 对应版本安装才是正解。

查询版本对应关系

一、环境

可以选择 Anaconda 软件包,也可以使用 Python 新建一个虚拟环境。

  • Anconda 安装
  • Python 虚拟环境

二、CUDA

1、CUDN 官网

就像 起序 中说到,当前最新版的 Tensorflow(2.1.0 or 2.2.0) 和最新版的 CUDA Toolkit 10.2 一起是会报错的,报错内容就在 起序 的下一行。

下载的是 CUDA Toolkit 10.1(Feb 2019),Online Documentation
【Tensorflow】Tensorflow 2.2.0 安装教程_第1张图片

如果下载的慢,可以考虑使用迅雷下载。

【Tensorflow】Tensorflow 2.2.0 安装教程_第2张图片

2、安装

默认即可

【Tensorflow】Tensorflow 2.2.0 安装教程_第3张图片
【Tensorflow】Tensorflow 2.2.0 安装教程_第4张图片

正式开始

【Tensorflow】Tensorflow 2.2.0 安装教程_第5张图片

同意并继续

【Tensorflow】Tensorflow 2.2.0 安装教程_第6张图片

选择自定义

【Tensorflow】Tensorflow 2.2.0 安装教程_第7张图片

在CUDA选项卡下,取消勾选 Visual Studio Intergration一项。(这是因为我们并没有使用Visual Studio开发环境)在 Driver Components 选项卡下,比较当前版本信息和安装版本信息。若当前版本高于安装版本,则取消勾选 Display Driver ;若当前版本低于或等于安装版本,保留默认信息安装即可。

【Tensorflow】Tensorflow 2.2.0 安装教程_第8张图片

下一步

【Tensorflow】Tensorflow 2.2.0 安装教程_第9张图片
【Tensorflow】Tensorflow 2.2.0 安装教程_第10张图片
【Tensorflow】Tensorflow 2.2.0 安装教程_第11张图片

查看是否安装成功

nvcc -V

【Tensorflow】Tensorflow 2.2.0 安装教程_第12张图片

三、CUDNN

1、CNDNN 官网

下载 CUDNN 需要注册一个账户,注册结束并登陆后才能下载。

【Tensorflow】Tensorflow 2.2.0 安装教程_第13张图片

点击 Downliad cnDNN 后会跳转到下一个 登陆/注册 界面,因为我已经注册过了,就不演示怎么注册了。

【Tensorflow】Tensorflow 2.2.0 安装教程_第14张图片
【Tensorflow】Tensorflow 2.2.0 安装教程_第15张图片

勾选 I Agree To the Terms of the cuDNN Software License Agreement , 然后选择对应 CUDA 下载的版本,因为 CUDA 选择的是 10.1,所以这里也是选择下面红框圈出的。

【Tensorflow】Tensorflow 2.2.0 安装教程_第16张图片
【Tensorflow】Tensorflow 2.2.0 安装教程_第17张图片

2、安装

此安装非彼安装,因为下载的 CUDNN 是一个压缩包,只需要在解压后,将整个文件复制到 CUDA 安装的目录内,为了简便,将文件的名字从 cudnn-10.1-windows10-x64-v7.6.5.32 更改为 cudnn765 即可。

至于为什么更改名字,因为 步骤四 需要 配置环境变量,文件名太长就太丑了。

【Tensorflow】Tensorflow 2.2.0 安装教程_第18张图片
【Tensorflow】Tensorflow 2.2.0 安装教程_第19张图片

四、配置环境变量

将自己安装的 CUDACUDNN 的目录配置到 path 中,如果和我目录差不多,就直接复制吧,省时间。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\cudnn765\cuda\bin

【Tensorflow】Tensorflow 2.2.0 安装教程_第20张图片

五、下载 tensorflow

进入 Python 虚拟环境

pip install tensorflow

六、卸载

如果想要卸载旧版本的,将下面图片红框内,带 版本号 的卸载即可,卸载时千万注意不要点错,一失足成千古恨。

【Tensorflow】Tensorflow 2.2.0 安装教程_第21张图片
如果对您有帮助,点个赞再走吧。

你可能感兴趣的:(tenorflow)