ติดตั้ง Tensorflow-GPU บน WSL2

Question

มีใครติดตั้ง Tensorflow-GPU บน WSL2 ด้วย NVIDIA GPU ได้สำเร็จบ้าง? ฉันมี Ubuntu 18.04 บน WSL2 แต่ฉันกำลังดิ้นรนเพื่อติดตั้งไดรเวอร์ NVIDIA ความช่วยเหลือใด ๆ จะได้รับการชื่นชมเมื่อฉันหลงทาง

Answer 1

ฉันสามารถยืนยันได้ว่าฉันสามารถทำงานได้โดยไม่ต้องใช้ Docker บน WSL2 ด้วยบทความต่อไปนี้:

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

อย่าลืมอัปเดตเป็นเวอร์ชันของไดรเวอร์460.15ไม่ใช่455.41ตามที่ระบุไว้ในเอกสาร CUDA

หมายเหตุสิ่งนี้ใช้ไม่ได้กับการ์ดในโหมด TCC (เฉพาะ WDDM) นอกจากนี้อย่าลืมวางไฟล์ของคุณบนระบบไฟล์ Linux (เช่นไม่ได้อยู่ในเมานต์ไดรฟ์เช่น/mnt/c/) ประสิทธิภาพจะเร็วขึ้นอย่างมากในระบบไฟล์ Linux (ซึ่งเกี่ยวข้องกับความแตกต่างในการนำ WSL 1 ไปใช้กับ WSL 2 ดู1 , 2และ3 )

หมายเหตุ:โปรดดูที่เธรดตัวสร้างคลาส (สืบทอดลำดับ) ใน Keras / Tensorflow หรือไม่