在Anaconda中安装用于TensorFlow v2.12的cuDNN时未找到匹配的发行版

0yg35tkg  于 2023-04-12  发布在  其他
关注(0)|答案(1)|浏览(174)

我尝试在Anaconda中安装TensorFlow v2.12,使用Python 3.9.16,在Windows 10中安装pip v23.0.1**。我的GPU(RTX 4080)需要Tensforflow v2.12,只有这个版本才能与我的GPU配合使用,因为支持Cuda Toolkit v11.8,这是最早支持Ada Lovelace GPU的版本。
Windowstensorflow.org操作系统的官方www.example.com提供了以下安装说明。

  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安装完毕,没有任何问题。但是,如果我尝试cuDNN8.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 www.example.com的想法8.6.0.163?
我尝试用以下方法安装cuDNN8.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,但这并不能解决问题。
在所有这些之后,我尝试8.6.0.163在Windows环境中安装cuDNN www.example.com版本,并且我设置了适当的环境变量,但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 compiler driver Copyright(c)2005-2022 NVIDIA Corporation Built on Wed_Sep_21_10:41:10_Pacific_Daylight_Time_2022 Cuda compilation tools,release 11.8,V11.8.89 Build cuda_11.8.r11.8/compiler.31833905_0

omhiaaxx

omhiaaxx1#

这里的问题是你试图在Windows操作系统中使用TensorFlow 2.12支持GPU。但是TensorFlow在本机Win操作系统中的TF 2.10版本之后停止了GPU支持,根据这个doc,它提到:
TensorFlow 2.10是在原生Windows上支持GPU的最后一个TensorFlow版本。从TensorFlow 2.11开始,您需要在WSL 2中安装TensorFlow,或者安装tensorflow-cpu,并可选地尝试TensorFlow-DirectML-Plugin
因此,请通过降级TensorFlow 2.10并安装兼容的CUDA 11.2cuDNN 8.1(如本测试构建配置中所述)或使用WSL 2directML插件在您的系统中通过Tensorflow获得GPU支持来重试。

相关问题