Как получить доступ к cuDNN в CuPy?

Привет, ребята! Я прохожу курс “от 0 до Linux админа” на https://yodo.im/courses/linux/?v=1d20b5ff1ee9 и столкнулся с проблемой при работе с CuPy и CuDNN.

Я установил пакеты cupy и cudnn с помощью uv:

$ uv init project
$ cd project
$ uv add cupy-cuda11x
$ uv run python -m cupyx.tools.install_library --cuda 11.x --library cudnn

Последние команды взяты из документации cuPy. CuDNN был установлен в /home/user/.cupy/cuda_lib/11.x/cudnn/8.8.1, и я увидел папки include и lib, содержащие все необходимые файлы CuDNN.

Пакет cupy-cuda11x работает – я проверил это с помощью Python-скриптов, импортируя CuPy. Но как мне проверить, что библиотека CuDNN работает с CuPy? Я пробовал:

import cupy.cudnn

или

from cupyx.cudnn import *

Но оба варианта вызывают ошибку:

  File "cupyx/cudnn.pyx", line 1, in init cupyx.cudnn
ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory

У меня установлена система Ubuntu 24.04 с CUDA 11.4. Процесс установки можно найти здесь.

Буду признателен за советы по этой проблеме! Как мне правильно подключить CuDNN к CuPy?

Спасибо за ответ! У меня все получилось, и я смог получить доступ к cuDNN в CuPy. Очень признателен за помощь!