Привет, ребята! Я прохожу курс “от 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?