Tensorflow не обнаруживает GPU - lambdalabs

Я пытаюсь запустить графический процессор tensorflow в виртуальной среде (venv):

я использую лямбдалабы ОС — Ubuntu 20.04.3 LTS.

У меня есть следующий скрипт Python: checkGPY.py:

импортировать тензорный поток как tf

если tf.test.gpu_device_name():
    print('Устройство GPU по умолчанию: {}'.format(tf.test.gpu_device_name()))
еще:
    print("Пожалуйста, установите версию TF для графического процессора")

Вне venv работает нормально. Я получаю устройство GPU по умолчанию: /device:GPU:0. Если обучить небольшую нейронную сеть (НС) и смотреть nvidia-smi Я вижу, что память графического процессора увеличивается во время обучения. Таким образом, ресурсы графического процессора используются для обучения НС.

Однако, если я запускаю его внутри venv (я установил версию tensorflow: 2.6.0 внутри venv.)

(venv) x@y $ python checkGPU.py

Я получаю: Пожалуйста, установите версию TF для графического процессора.

Я также получаю следующее: Не удалось загрузить динамическую библиотеку 'libcudnn.so.8'; dlerror: libcudnn.so.8: невозможно открыть общий объектный файл: нет такого файла или каталога

Я так понимаю, что динамическая библиотека libcudnn.so.8 недоступна изнутри venv.

Как я могу решить эту проблему?