如何使用较低（兼容）版本的 cudatoolkit 和 cudnn 安装 tensorflowgpu-version

Question

每次我在 anaconda 上安装 tensorflow-gpu 时，它都会自动安装 tensorflow-gpu cudatoolkit 和 cudnn 的最新可用版本，即使它与图形不兼容 我已经在我的计算机上安装了卡和 cuda，这在以后造成了一个大问题 模拟。

例如我有RTX3060，我正在安装tensorflow-gpu 2.5.0，我已经安装了 cuda 11.1 但是当我使用 anaconda 安装它时

"conda install tensorflow-gpu=2.5.0" 

它将安装cudatoolkit 11.3.x...和最先进的可用cudnn

Answer 1

您可以使用兼容的cudatoolkit 和cudnn 版本使用以下命令安装tensorflow-gpu。

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
python3 -m pip install tensorflow-gpu==2.5.0
# Verify install:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

有关分步说明，请参阅此link。