Không thể cài đặt cuda và cuDNN đúng cách

Tôi gặp khó khăn lớn khi cài đặt cuda và cuDNN để có thể sử dụng GPU của mình với thư viện máy học.

TLDR; Giúp tôi cài đặt cuda và cuDNN để hoạt động với cấu hình của tôi.

Phiên bản dài Cuối cùng tôi cũng thành công khi thấy gpu của mình với

tf.config.list_physical_devices('GPU').

Nhưng khi tôi muốn biên dịch một mô hình, tôi gặp lỗi sau:

Đã tải thư viện CuDNN thời gian chạy: 8.0.5 nhưng nguồn được biên dịch bằng: 8.1.0. Thư viện CuDNN cần phải có phiên bản chính phù hợp và phiên bản phụ bằng hoặc cao hơn. Nếu sử dụng cài đặt nhị phân, hãy nâng cấp thư viện CuDNN của bạn. Nếu xây dựng từ các nguồn, hãy đảm bảo rằng thư viện được tải trong thời gian chạy tương thích với phiên bản được chỉ định trong quá trình cấu hình biên dịch.

Tôi đã cài đặt cuda 11.2 và cuDNN 8.1 nhưng tệp của cuDNN là 8.0.5. Tôi không hiểu tại sao, vì vậy tôi đã xóa cuda và cuDNN. Làm thế nào tôi có thể cài đặt phiên bản tốt.

Cấu hình

Máy tính: bộ xử lý amd, RTX 3060, phiên bản trình điều khiển nvidia: 495.44, ubuntu 20.04

Trăn 3.9.5

tensorflow==2.6.0 & tensorflow-gpu==2.6.0