Cài đặt Tensorflow-GPU trên WSL2

Question

Có ai đã cài đặt thành công Tensorflow-GPU trên WSL2 với GPU NVIDIA không? Tôi có Ubuntu 18.04 trên WSL2, nhưng đang đấu tranh để cài đặt trình điều khiển NVIDIA. Mọi sự giúp đỡ sẽ được đánh giá cao vì tôi bị mất.

Answer 1

Tôi có thể xác nhận rằng tôi có thể làm cho điều này hoạt động mà không cần Docker trên WSL2 nhờ vào bài viết sau:

https://qiita.com/Navier/items/cf551908bae707db4258

Đảm bảo cập nhật lên phiên bản trình điều khiển 460.15, không 455.41được liệt kê trong tài liệu CUDA.

Lưu ý, điều này không hoạt động với thẻ ở chế độ TCC (chỉ WDDM). Ngoài ra, hãy đảm bảo đặt các tệp của bạn trên hệ thống tệp Linux (tức là không phải trên ổ đĩa gắn kết chẳng hạn /mnt/c/). Hiệu suất nhanh hơn đáng kể trên hệ thống tệp Linux (điều này liên quan đến sự khác biệt trong việc triển khai WSL 1 so với WSL 2; xem 1 , 2 và 3 ).

LƯU Ý: Xem thêm Luồng trình tạo lớp (kế thừa Trình tự) có an toàn trong Keras / Tensorflow không?