升级到cudnn 9.1.1，但Python只识别8.5.0

Question

我在 Jupyter Lab 工作，由 PyCharm 管理。我最近将我的 cuda 升级到 12.4：

which nvcc

产量：

/usr/local/cuda-12.4/bin/nvcc

我还使用 x86_64 ubuntu 2204 本地 deb 安装方法将 cudnn 升级到 9.1.1 https://developer.nvidia.com/cudnn-downloads 不幸的是，当我尝试时（在 Jupyter Lab 下的 .ipynb 笔记本中）：

import torch
cudnn_version = torch.backends.cudnn.version()
print("CuDNN version:", cudnn_version)

我得到：

CuDNN version: 8500 

附加信息： 软件和更新表明 GeForce GTX 1080 Ti，使用来自 nvidia-driver-550（专有）的 NVIDIA 驱动程序元包

有人有解决这个问题的建议吗？预先感谢。

Answer 1

卸载 Tensorflow 和 Pytorch，重新启动并重新安装它们（通过 Pycharm）就可以了。