我尝试使用Python 3.9.16在 Anaconda 中安装 TensorFlow v2.12 ,并使用pip v23.0.1在 Windows 10操作系统中安装 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.0python3 -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 …Run Code Online (Sandbox Code Playgroud) 我已经在Windows 10中安装了WSL2 (Ubuntu 22.04 Kernel)、Tensorflow 2.12、Cuda Toolkit 11.8.0和cuDNN 8.6.0.163在 Miniconda 环境 (Python 3.9.16) 中,通常按照官方的tensorflow.org 推荐进行安装。我应该强调的是,我想使用 Tensorflow 2.12,因为使用相应的 Cuda Toolkit 11.8.0,它与 Ada Lovelace GPU(我的情况是 RTX4080)兼容。
当我去训练我的模型时,它给出了以下错误:
"Loaded cuDNN version 8600 Could not load library libcudnn_cnn_infer.so.8. Error: libcuda.so : cannot open shared object file: No such file or directory".
Run Code Online (Sandbox Code Playgroud)
有没有什么想法出了问题*?
路径配置如下:
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
echo 'CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
Run Code Online (Sandbox Code Playgroud)
使用以下命令搜索引用我的错误的文件:
ldconfig -p | grep libcudnn_cnn …