Tensorflow không phát hiện GPU - lambdalabs

Tôi cố gắng thiết lập và chạy GPU tensorflow trong môi trường ảo (venv):

tôi sử dụng lambdalabs Hệ điều hành là Ubuntu 20.04.3 LTS.

Tôi có tập lệnh python sau: checkGPY.py:

nhập tenorflow dưới dạng tf

nếu tf.test.gpu_device_name():
    print('Thiết bị GPU mặc định: {}'.format(tf.test.gpu_device_name()))
khác:
    print("Vui lòng cài đặt phiên bản GPU của TF")

Bên ngoài venv nó hoạt động tốt. Tôi nhận được Thiết bị GPU mặc định: /device:GPU:0. Nếu đào tạo một mạng lưới thần kinh nhỏ (NN) và xem nvidia-smi Tôi thấy rằng bộ nhớ GPU tăng lên trong quá trình đào tạo. Vì vậy, tài nguyên GPU được sử dụng để đào tạo NN.

Tuy nhiên, nếu tôi chạy thì nó nằm trong venv (Tôi đã cài đặt phiên bản tensorflow: 2.6.0 bên trong venv.)

(venv) x@y $ python checkGPU.py

Tôi nhận được: Vui lòng cài đặt phiên bản GPU của TF

Tôi cũng nhận được thông tin sau: Không thể tải thư viện động 'libcudnn.so.8'; dlerror: libcudnn.so.8: không thể mở tệp đối tượng được chia sẻ: Không có tệp hoặc thư mục như vậy

Vì vậy, tôi hiểu rằng không thể truy cập thư viện động libcudnn.so.8 từ bên trong venv.

Làm thế nào tôi có thể giải quyết điều này?