在 Anaconda 中安装用于 TensorFlow v2.12 的 cuDNN 时未找到匹配的分布

问题描述 投票:0回答:0

我尝试在 Anaconda 中使用 Python 3.9.16Windows 10 OS 以及 pip v23.0.1 安装 TensorFlow v2.12。我的 GPU (RTX4080) 需要 Tensforflow v2.12,并且只有这个版本适用于我的 GPU,因为它支持 Cuda Toolkit v11.8,这是支持 Ada Lovelace GPU 的最老版本。

适用于 Windows 操作系统的官方 tensorflow.org 提供以下安装说明。

  1. conda install -c conda-forge cudatoolkit=11.8.0
  2. python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.*
  3. CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
  4. export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib

验证安装:

  1. python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

以上步骤,Cuda Toolkit 11.8.0Tensorflow 2.12安装没有任何问题。但是,如果我尝试安装 cuDNN 8.6.0.163,没有安装,并给出以下结果。

Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com
ERROR: Could not find a version that satisfies the requirement nvidia-cudnn-cu11==8.6.0.163 (from versions: 0.0.1.dev5, 2021.10.26, 2021.11.18, 2021.12.8, 2022.1.13, 2022.4.2, 2022.5.19)
ERROR: No matching distribution found for nvidia-cudnn-cu11==8.6.0.163

怎么了?有没有关于如何安装 cuDNN 8.6.0.163 的想法?

我尝试通过以下方式安装 cuDNN 8.6.0.163 并且都给出了相同的结果:

python -m pip install nvidia-cudnn-cu11==8.6.0.163
(我的conda环境中的python是v3.9.16所以python3命令不存在,它与python相同。我已经检查过:
python --version

pip install nvidia-cudnn-cu11==8.6.0.163

python -m pip install nvidia-cudnn-cu11==8.6.0.163 --no-cache-dir

python -m pip install nvidia-cudnn-cu11==8.6.0.163 --proxy="xxx.xxx.xxx.xxx:xxxx"
(其中xxx.xxx.xxx.xxx:xxxx一些代理IP和对应的端口)

此外,我安装了'nvidia-pyindex':

pip install nvidia-pyindex
,但这并不能解决问题。

在这一切之后,我尝试在 Windows 环境下安装 cuDNN 8.6.0.163 版本,并设置了适当的环境变量,但是 Anaconda 中的 Tenrosflow 没有看到,等等:

print(tf.config.list_physical_devices('GPU'))

什么都不返回:[ ]

从第 3 步开始,我检查了这个:

python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)

返回的地方:

Traceback (most recent call last):
  File "<string>", line 1, in <module>
ModuleNotFoundError: No module named 'nvidia'

在 PowerShell 中:

nvidia-smi

返回这个:

“NVIDIA-SMI 531.41 | 驱动程序版本:531.41 | CUDA 版本:12.1”

nvcc -V

返回这个:

nvcc: NVIDIA (R) Cuda 编译器驱动 版权所有 (c) 2005-2022 NVIDIA 公司 建立于 Wed_Sep_21_10:41:10_Pacific_Daylight_Time_2022 Cuda 编译工具,版本 11.8,V11.8.89 构建 cuda_11.8.r11.8/compiler.31833905_0

python-3.x tensorflow anaconda cudnn
© www.soinside.com 2019 - 2024. All rights reserved.