ฉันกำลังพยายามตั้งค่าเครื่อง Lenovo Legion 5 Pro ด้วย Ubuntu 20.04.3 LTS เพื่อให้ฉันสามารถใช้ GPU (Nvidia Geforce RTX 3070 8GB) สำหรับการเรียนรู้ของเครื่อง เช่น กับ PyTorch อย่างไรก็ตาม แม้ว่าหลังจากติดตั้งไดรเวอร์ NVidia ใหม่หลายครั้ง ดูเหมือนว่าจะยังมองไม่เห็น เนื่องจากการรัน nvidia-smi จะแสดงสิ่งต่อไปนี้:
NVIDIA-SMI ล้มเหลวเนื่องจากไม่สามารถสื่อสารกับ NVIDIA ได้
คนขับรถ ตรวจสอบให้แน่ใจว่าได้ติดตั้งและเรียกใช้ไดรเวอร์ NVIDIA ล่าสุดแล้ว
ฉันได้ดูวิธีแก้ปัญหาหลายอย่างที่นั่น (นี้, นี้, นี้, นี้) และไม่มีอะไรทำงานจนถึงตอนนี้ แต่ฉันคิดว่าปัญหาเกิดจากเคอร์เนลที่ฉันใช้
วิ่ง nvcc -- รุ่น ให้ผลลัพธ์ต่อไปนี้:
nvcc: ไดรเวอร์คอมไพเลอร์ NVIDIA (R) Cuda
ลิขสิทธิ์ (c) 2005-2019 NVIDIA Corporation
สร้างเมื่อ Sun_Jul_28_19:07:16_PDT_2019
เครื่องมือรวบรวม Cuda, รีลีส 10.1, V10.1.243
ls /usr/src | grep nvidia
เอาต์พุต nvidia-470.86
เมื่อฉันพยายามเรียกใช้คำสั่งเหล่านี้จาก ที่นี่, เช่น.
sudo apt ติดตั้ง dkms
sudo dkms ติดตั้ง -m nvidia -v 470.86
ฉันได้รับข้อผิดพลาดต่อไปนี้:
ข้อผิดพลาด! ไม่พบส่วนหัวเคอร์เนลของคุณสำหรับเคอร์เนล 5.16.2-051602-generic
โปรดติดตั้งแพ็คเกจ linux-headers-5.16.2-051602-generic
หรือใช้ตัวเลือก --kernelsourcedir เพื่อบอก DKMS ว่าอยู่ที่ไหน
ตอนนี้, linux-headers-5.16.2-051602-ทั่วไป
ไม่มีอยู่แต่ฉันมีแล้ว ลินุกซ์ส่วนหัวทั่วไป
ติดตั้งแล้ว
คุณมีข้อเสนอแนะเกี่ยวกับสิ่งที่ฉันสามารถแก้ไขได้หรือไม่
ขอบคุณมาก!