ไม่เป็นทางการ แต่คุณสามารถเปลี่ยนได้ การกระจาย
ตัวแปรบน หน้าคำแนะนำ เข้าไปข้างใน อูบุนตู20.04
, แบบนี้:
การกระจาย = 'ubuntu20.04' \
&& ขด -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key เพิ่ม - \
&& ขด -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo ที /etc/apt/sources.list.d/nvidia-docker.list
ส่วนที่เหลือเหมือนกัน:
sudo apt-get อัปเดต
sudo apt-get install -y nvidia-docker2
sudo systemctl รีสตาร์ทนักเทียบท่า
จากนั้น คุณสามารถตรวจสอบการติดตั้งของคุณ:
sudo docker run --rm --gpus ทั้งหมด nvidia/cuda:11.0-base nvidia-smi
ควรส่งคืนสิ่งนี้:
+--------------------------------------------- ----------------------------+
| เวอร์ชันไดรเวอร์ NVIDIA-SMI 450.51.06: 450.51.06 เวอร์ชัน CUDA: 11.0 |
|---------------------------------------++---------------- ----------------------------+
| ชื่อ GPU ความคงทน-M| Bus-Id Disp.A | ผันผวน Uncorr ECC |
| Fan Temp Perf Pwr:การใช้งาน/ความจุ| การใช้งานหน่วยความจำ | GPU-Util Compute M. |
| | | MIG M. |
|==============================+================== =====+======================|
| 0 เทสลา T4 เปิด | 00000000:00:1E.0 ปิด | 0 |
| ไม่ระบุ 34C P8 9W / 70W | 0MiB / 15109MiB | ค่าเริ่มต้น 0% |
| | | ไม่มี |
+--------------------------------------+-------- ----------------------------+
+--------------------------------------------- ----------------------------+
| กระบวนการ: |
| GPU GI CI ประเภท PID ชื่อกระบวนการ หน่วยความจำ GPU |
| รหัส ID การใช้ |
|================================================= ============================|
| ไม่พบกระบวนการทำงาน |
+--------------------------------------------- ----------------------------+
NB
ฉันแค่ต้องใช้ nvidia-นักเทียบท่า
เพื่อทำการเรียนรู้เชิงลึกด้วย tensorflow และวิธีแก้ปัญหาที่ฉันให้ไว้ข้างต้นก็ไม่มีปัญหาสำหรับการฝึกอบรมและการอนุมาน