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

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

Я использую lambdalabs ОС — Ubuntu 20.04.3 LTS.

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

import tensorflow as tf

if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")

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

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

(venv) x@y $ python checkGPU.py

Я получаю: Пожалуйста, установите версию GPU из TF

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

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

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