我尝试在 Anaconda 中使用 Python 3.9.16 和 Windows 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 提供以下安装说明。
conda install -c conda-forge cudatoolkit=11.8.0
python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.*
CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
以上步骤,Cuda Toolkit 11.8.0和Tensorflow 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