ฉันมีปัญหาใหญ่ในการติดตั้ง cuda และ cuDNN เพื่อให้สามารถใช้ GPU กับไลบรารีแมชชีนเลิร์นนิงได้
TLDR;
ช่วยฉันติดตั้ง cuda และ cuDNN เพื่อทำงานกับการกำหนดค่าของฉัน
รุ่นยาว
ในที่สุดฉันก็ประสบความสำเร็จเห็น GPU ของฉันด้วย
tf.config.list_physical_devices('GPU')
.
แต่เมื่อฉันต้องการรวบรวมแบบจำลองฉันมีข้อผิดพลาดดังต่อไปนี้:
โหลดไลบรารี CuDNN รันไทม์: 8.0.5 แต่ซอร์สถูกคอมไพล์ด้วย: 8.1.0 ไลบรารี CuDNN ต้องมีเวอร์ชันหลักที่ตรงกันและเวอร์ชันรองที่เท่ากันหรือสูงกว่า หากใช้การติดตั้งแบบไบนารี ให้อัปเกรดไลบรารี CuDNN ของคุณ หากสร้างจากแหล่งที่มา ตรวจสอบให้แน่ใจว่าไลบรารีที่โหลดขณะรันไทม์เข้ากันได้กับเวอร์ชันที่ระบุระหว่างการกำหนดค่าการคอมไพล์
ฉันติดตั้ง cuda 11.2 และ cuDNN 8.1 แล้ว แต่ไฟล์ของ cuDNN เป็น 8.0.5 ฉันไม่เข้าใจว่าทำไม ฉันจึงกำจัด cuda และ cuDNN ฉันจะติดตั้งเวอร์ชันที่ดีได้อย่างไร
การกำหนดค่า
คอมพิวเตอร์: โปรเซสเซอร์ amd, RTX 3060, ไดรเวอร์ nvidia เวอร์ชัน: 495.44, ubuntu 20.04
ไพธอน 3.9.5
tensorflow==2.6.0 & tensorflow-gpu==2.6.0