Tensorflow 2.2 GPU-要安装哪个cuDNN库?

Tensorflow 2.2 GPU-要安装哪个cuDNN库?,tensorflow,cudnn,Tensorflow,Cudnn,我已经成功安装了CUDA驱动程序、cuDNN库和tensorflow。但是当运行一个只导入tensorflow的测试程序时,我会得到一个错误。该错误似乎表明我安装了错误版本的cuDNN库。我希望能在这方面得到一些帮助。如果我需要降级cuDNN,我该怎么做 Tensorflow版本:2.2 GPU 操作系统:Ubuntu 16.04.6 LTS(GNU/Linux 4.4.0-184-generic x86_64) nvcc-V显示以下信息: nvcc -V nvcc: NVIDIA (R) Cu

我已经成功安装了CUDA驱动程序、cuDNN库和tensorflow。但是当运行一个只导入tensorflow的测试程序时,我会得到一个错误。该错误似乎表明我安装了错误版本的cuDNN库。我希望能在这方面得到一些帮助。如果我需要降级cuDNN,我该怎么做

Tensorflow版本:2.2 GPU 操作系统:Ubuntu 16.04.6 LTS(GNU/Linux 4.4.0-184-generic x86_64) nvcc-V显示以下信息:

nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Tue_Aug_11_14:27:32_CDT_2015
Cuda compilation tools, release 7.5, V7.5.17
Fri Jun 12 17:16:38 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.36.06    Driver Version: 450.36.06    CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  GeForce GTX 980 Ti  Off  | 00000000:02:00.0 Off |                  N/A |
| 22%   27C    P8    17W / 250W |     74MiB /  6083MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1489      G   /usr/lib/xorg/Xorg                 71MiB |
+-----------------------------------------------------------------------------+
nvidia smi显示以下信息:

nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Tue_Aug_11_14:27:32_CDT_2015
Cuda compilation tools, release 7.5, V7.5.17
Fri Jun 12 17:16:38 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.36.06    Driver Version: 450.36.06    CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  GeForce GTX 980 Ti  Off  | 00000000:02:00.0 Off |                  N/A |
| 22%   27C    P8    17W / 250W |     74MiB /  6083MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1489      G   /usr/lib/xorg/Xorg                 71MiB |
+-----------------------------------------------------------------------------+
cuDNN按照说明成功安装,但我想我已经安装了11.0版

程序尝试导入tensorflow(python 3.6)时出现错误消息


按照以下步骤,对于tensorflow 2.2,您需要CUDA 10.1和cuDNN 7.4:

CUDA存档/旧版本:

cuDNN存档,您必须使用nvidia帐户才能访问:

特别值得注意的是,7.4版本中没有与10.1兼容的cuDNN,所以我会尝试7.5.0。安装cuDNN只需将下载的文件复制到安装CUDA的文件夹中(在各自的文件夹中)