ฉันจะติดตั้งไดรเวอร์ NVIDIA และ CUDA ลงใน Ubuntu ได้อย่างไร


33

ฉันมีระบบที่มีการ์ดของ NVIDIA นั้นจะมีการสนับสนุนการคำนวณของ 3.5+ เทียบบนhttps://developer.nvidia.com/cuda-gpus ฉันจะติดตั้ง CUDA และไดรเวอร์ NVIDIA ใน Ubuntu โดยไม่ต้องดาวน์โหลดไฟล์. deb จาก NVIDIA ได้อย่างไร

คำตอบ:


52

Ubuntu 18.04, CUDA 10.1, ไดรเวอร์ libcudnn 7.5.1 และ NVIDIA 418.67:

หมายเหตุ 2019-06-23:อัพเดตล่าสุดด้วยไดรเวอร์ CUDA 10.0 หรือ 10.1 เวอร์ชัน NVIDIA 418.67 ที่ติดตั้งพร้อมไม่มีไลบรารี 32 บิตรวมอยู่อีกต่อไปซึ่งจะทำให้ Steam และเกมส่วนใหญ่ไม่ทำงานอีกต่อไป เวอร์ชั่นlibnvidia-gl-418:i386ติดตั้งรุ่น 418.56 เท่านั้นซึ่งจะไม่ทำงานกับไดรเวอร์ 418.67 หวังว่า NVIDIA จะวางจำหน่ายการอัปเดตในเร็ว ๆ นี้ ฉันได้เพิ่มข้อมูลที่ด้านล่างของคำตอบนี้ในrun file installส่วนของวิธีการดาวน์โหลดเฉพาะไฟล์รันสำหรับตัวติดตั้ง CUDA จากนั้นคุณสามารถใช้ไดรเวอร์ใดก็ได้ที่คุณต้องการ ไฟล์ run มีขนาด 2.3GB ดังนั้นอาจต้องใช้เวลาดาวน์โหลดสักหน่อย

หมายเหตุ: CUDA 9.x ไม่สามารถใช้งานได้ผ่าน ubuntu1804 repo ของ NVIDIA อย่างไรก็ตามฉันเขียนคำตอบสำหรับ CUDA 9.2 ที่ https://askubuntu.com/a/1086993/231142

คำแนะนำเหล่านี้สำหรับการติดตั้ง CUDA ผ่านที่เก็บข้อมูลแทนการ.debติดตั้ง

บรรทัดต่อไปนี้คุณสามารถคัดลอกและวางไปที่หน้าต่างเทอร์มินัล กดCtrl+ Alt+ Tเพื่อเปิดหน้าต่างเทอร์มินัล

ลบ CUDA PPA ใด ๆ ที่อาจติดตั้งและลบnvidia-cuda-toolkitหากติดตั้งไว้:

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove --autoremove nvidia-cuda-toolkit

แนะนำให้นำไดรเวอร์ NVIDIA ทั้งหมดออกก่อนติดตั้งไดรเวอร์ใหม่:

sudo apt remove --autoremove nvidia-*

จากนั้นอัพเดตระบบ:

sudo apt update

เมื่อเร็ว ๆ นี้ฉันเพิ่งพบว่าการติดตั้ง CUDA นั้นทำงานได้graphics-drivers ppaดังนั้นถ้าคุณยังไม่ได้เพิ่มให้เพิ่มทันที:

sudo add-apt-repository ppa:graphics-drivers/ppa

ติดตั้งกุญแจ:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

เพิ่ม repos:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda_learn.list'

อัปเดตระบบอีกครั้ง:

sudo apt update

ติดตั้ง CUDA 10.1:

sudo apt install cuda-10-1

มันควรจะติดตั้งไดรเวอร์ NVIDIA 418.40 พร้อมกับสิ่งที่ระบุไว้ใน repo ดู: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/

ติดตั้ง libcudnn7 7.5.1:

sudo apt install libcudnn7

เพิ่มบรรทัดต่อไปนี้ลงใน~/.profileไฟล์ของคุณสำหรับ CUDA 10.1

# set PATH for cuda 10.1 installation
if [ -d "/usr/local/cuda-10.1/bin/" ]; then
    export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

รีบูทคอมพิวเตอร์และตรวจสอบการตั้งค่าของคุณเมื่อรีบูตเสร็จสมบูรณ์:

ตรวจสอบ NVIDIA Cuda Compiler ด้วยnvcc --version:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Apr_24_19:10:27_PDT_2019
Cuda compilation tools, release 10.1, V10.1.168

ตรวจสอบเวอร์ชัน libcudnn /sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn:

terrance@terrance-ubuntu:~$ /sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn
    libcudnn.so.7 -> libcudnn.so.7.5.1

ตรวจสอบไดรเวอร์ NVIDIA ด้วยnvidia-smi:

terrance@terrance-ubuntu:~$ nvidia-smi 
Sat Jun  1 09:38:07 2019       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.67       Driver Version: 418.67       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 750 Ti  On   | 00000000:02:00.0  On |                  N/A |
| 40%   38C    P0     2W /  38W |    116MiB /  2000MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      2216      G   /usr/lib/xorg/Xorg                           110MiB |
|    0      2542      G   compton                                        1MiB |
+-----------------------------------------------------------------------------+

.run ติดตั้งไฟล์

โดยการใช้งานsudo add-apt-repository ppa:graphics-drivers/ppaคุณสามารถติดตั้ง430.26ไดรเวอร์ใหม่ล่าสุดหรือสิ่งที่เหมาะกับความต้องการของคุณ

ถัดไปติดตั้ง libcudnn7 โดยทำตาม:

เพิ่ม Repo:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda_learn.list'

ติดตั้งกุญแจ:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

อัปเดตระบบ:

sudo apt update

ติดตั้ง libcudnn7.5.1:

sudo apt install libcudnn7

ตอนนี้ดาวน์โหลดcuda_10.1.105_418.39_linux.runจากhttps://developer.nvidia.com/cuda-10.1-download-archive-base?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal

จากนั้นรันโปรแกรมติดตั้ง:

sudo sh cuda_10.1.105_418.39_linux.run

พิมพ์ accept และกด enter บนหน้าจอนี้:

┌──────────────────────────────────────────────────────────────────────────────┐
│  End User License Agreement                                                  │
│  --------------------------                                                  │
│                                                                              │
│                                                                              │
│  Preface                                                                     │
│  -------                                                                     │
│                                                                              │
│  The Software License Agreement in Chapter 1 and the Supplement              │
│  in Chapter 2 contain license terms and conditions that govern               │
│  the use of NVIDIA software. By accepting this agreement, you                │
│  agree to comply with all the terms and conditions applicable                │
│  to the product(s) included herein.                                          │
│                                                                              │
│                                                                              │
│  NVIDIA Driver                                                               │
│                                                                              │
│                                                                              │
│  Description                                                                 │
│                                                                              │
│  This package contains the operating system driver and                       │
│──────────────────────────────────────────────────────────────────────────────│
│ Do you accept the above EULA? (accept/decline/quit):                         │
│ accept                                                                       

ยกเลิกการเลือกไดรเวอร์แล้วเลือกติดตั้งโดยใช้ปุ่มลูกศรและสเปซบาร์เพื่อย้ายและเลือกหรือยกเลิกการเลือก:

┌──────────────────────────────────────────────────────────────────────────────┐
│ CUDA Installer                                                               │
│ - [ ] Driver                                                                 │
│      [ ] 418.39                                                              │
│ + [X] CUDA Toolkit 10.1                                                      │
│   [X] CUDA Samples 10.1                                                      │
│   [X] CUDA Demo Suite 10.1                                                   │
│   [X] CUDA Documentation 10.1                                                │
│   Install                                                                    │
│   Options                                                                    │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│ Up/Down: Move | Left/Right: Expand | 'Enter': Select | 'A': Advanced options │

รอให้การติดตั้งเสร็จสิ้นอาจเกิดข้อผิดพลาดในระหว่างนั้น แต่ไม่ต้องกังวล

เพิ่มบรรทัดต่อไปนี้ลงใน~/.profileไฟล์ของคุณสำหรับ CUDA 10.1

# set PATH for cuda 10.1 installation
if [ -d "/usr/local/cuda-10.1/bin/" ]; then
    export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

รีบูตระบบเพื่อให้การเปลี่ยนแปลงมีผล


1
โปรดทราบว่าตั้งแต่ 2018-11 คุณจะต้อง cuda 0.9 สำหรับการดึงข้อมูล
mathtick

1
การติดตั้ง nvidia-driver-410 ล้มเหลวเพราะมันขึ้นอยู่กับ xserver-xorg-video-nvidia-410 และมันไม่สามารถติดตั้งได้เพราะมันขึ้นอยู่กับเครือข่าย lbnvidia ที่ไม่สามารถใช้งานได้
fccoelho

1
@ fccoelho ฉันเพิ่งค้นพบว่าการติดตั้งใช้งานได้ดีที่สุดด้วยgraphics-drivers ppaดังนั้นฉันจึงได้อัปเดตคำตอบของฉันที่นี่เพื่อรวม PPA นั้นจากนั้นก็ควรจะค้นหาและติดตั้งทั้งหมดขึ้นอยู่กับ
Terrance

1
คุณทำตามขั้นตอน "sudo apt autoremove" หลังจากลบชุดเครื่องมือและไดรเวอร์ก่อนหน้านี้ มิฉะนั้นการติดตั้งจะล้มเหลวเนื่องจากการอ้างอิงที่ไม่ได้รับการอัพเดตโดยอัตโนมัติ
Pieter-Jan Busschaert

1
ความคิดเห็นเพิ่มเติม: ไดรเวอร์ NVidia ที่รวมอยู่ในแพ็คเกจ 10.1 CUDA .deb จาก NVidia ไม่รวมไลบรารี 32- บิต Steam (และเกมส่วนใหญ่) จะไม่ทำงานอีกต่อไปหลังจากติดตั้งด้วยวิธีนี้ ดูเพิ่มเติมที่: github.com/ValveSoftware/steam-for-linux/issues/5778
Pieter-Jan Busschaert

8

ไดรเวอร์ Ubuntu 16.04, CUDA 9.2 และ NVIDIA 396:

หมายเหตุ: repo ของ NVIDIA ที่นี่ได้ตัดสินใจที่จะผลักดันไดรเวอร์ 410 ฉันจะทำการทดสอบเพื่อดูว่าฉันจะได้รับการตั้งค่าสำหรับไดรเวอร์ที่คุณต้องการติดตั้ง

คำแนะนำเหล่านี้สำหรับการติดตั้ง CUDA ผ่านที่เก็บข้อมูลแทนการ.debติดตั้ง

บรรทัดต่อไปนี้คุณสามารถคัดลอกและวางไปที่หน้าต่างเทอร์มินัล กดCtrl+ Alt+ Tเพื่อเปิดหน้าต่างเทอร์มินัล

ลบ CUDA PPA ใด ๆ ที่อาจติดตั้งและลบnvidia-cuda-toolkitหากติดตั้งไว้:

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove nvidia-cuda-toolkit

แนะนำให้นำไดรเวอร์ NVIDIA ทั้งหมดออกก่อนติดตั้งไดรเวอร์ใหม่:

sudo apt remove nvidia-*

จากนั้นอัพเดตระบบ:

sudo apt update

ติดตั้งกุญแจ:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

เพิ่ม repo:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

อัปเดตสำหรับ repo ใหม่ที่ถูกเพิ่ม:

sudo apt update

ติดตั้ง CUDA 9.2:

sudo apt install cuda-9-2

มันควรจะติดตั้งไดรเวอร์ nvidia-396 พร้อมกับสิ่งที่ระบุไว้ใน repo ดู: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/

ตรวจสอบว่าติดตั้ง CUDA 9.2 แล้ว:

~$ ls /usr/local/cuda-9.2/
bin     include    libnvvp           nvml    samples  targets
doc     lib64      LICENSE           nvvm    share    tools
extras  libnsight  nsightee_plugins  README  src      version.txt

ตอนนี้เพิ่มต่อไปนี้เพื่อคุณ~/.profileสำหรับและPATH LD_LIBRARYคุณสามารถใช้คำสั่งgedit ~/.profileเพื่อแก้ไข:

# set PATH for cuda 9.2 installation
if [ -d "/usr/local/cuda-9.2/bin/" ]; then
    export PATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

รีบูตระบบของคุณ

sudo reboot

เมื่อระบบหยุดทำงานคุณสามารถตรวจสอบการติดตั้งได้โดยพิมพ์ข้อความต่อไปนี้:

nvcc -V

คุณควรเห็นบางสิ่งที่คล้ายกันดังต่อไปนี้:

~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

และคุณจะเห็น396.xxไดรเวอร์ที่ติดตั้ง:

~$ nvidia-smi
Thu May 17 07:38:54 2018        
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 396.44                 Driver Version: 396.44                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 760     Off  | 00000000:02:00.0 N/A |                  N/A |
| 49%   53C    P0    N/A /  N/A |    187MiB /  1999MiB |     N/A      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0                    Not Supported                                       |
+-----------------------------------------------------------------------------+


3

ไดรเวอร์ Ubuntu 14.04, CUDA 10.0 และ NVIDIA 410:

โปรดทราบว่าฉันไม่ได้ติดตั้ง Ubuntu 14.04 ดังนั้นฉันไม่สามารถตรวจสอบว่าขั้นตอนเหล่านี้ทำงานได้หรือไม่ แต่ตามhttp://developer.download.nvidia.com/compute/cuda/repos/จาก NVIDIA มันควรเป็นขั้นตอนเดียวกับที่ฉันได้ระบุไว้ในอีกสองคำตอบ

คำแนะนำเหล่านี้สำหรับการติดตั้ง CUDA ผ่านที่เก็บข้อมูลแทนการ.debติดตั้ง

บรรทัดต่อไปนี้คุณสามารถคัดลอกและวางไปที่หน้าต่างเทอร์มินัล กดCtrl+ Alt+ Tเพื่อเปิดหน้าต่างเทอร์มินัล

ลบ CUDA PPA ใด ๆ ที่อาจติดตั้งและลบnvidia-cuda-toolkitหากติดตั้งไว้:

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove nvidia-cuda-toolkit

แนะนำให้นำไดรเวอร์ NVIDIA ทั้งหมดออกก่อนติดตั้งไดรเวอร์ใหม่:

sudo apt remove nvidia-*

จากนั้นอัพเดตระบบ:

sudo apt update

ติดตั้งกุญแจ:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/7fa2af80.pub

เพิ่ม repo:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

อัปเดตระบบอีกครั้ง:

sudo apt update

ติดตั้ง CUDA 10.0

sudo apt install cuda-10-0

มันควรจะติดตั้งไดรเวอร์ nvidia-410 พร้อมกับสิ่งที่ระบุไว้ใน repo ดู: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/

เพิ่มบรรทัดต่อไปนี้ลงใน~/.profileไฟล์ของคุณสำหรับ CUDA 10.0

# set PATH for cuda 10.0 installation
if [ -d "/usr/local/cuda-10.0/bin/" ]; then
    export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

รีบูทคอมพิวเตอร์และตรวจสอบการตั้งค่าของคุณเมื่อรีบูตเสร็จสมบูรณ์:

ตรวจสอบ NVIDIA Cuda Compiler ด้วยnvcc --version:

ตรวจสอบไดรเวอร์ NVIDIA ด้วยnvidia-smi:


0

อย่าทำผิด ดาวน์โหลดไฟล์. deb ครั้งแล้วครั้งเล่า ฉันดาวน์โหลดไฟล์. run และติดตั้ง แต่เมื่อฉันลองติดตั้ง tensorrt มันกลับกลายเป็นว่ามันสามารถติดตั้งได้ก็ต่อเมื่อระบบของคุณใช้การติดตั้งไฟล์. deb บนไดรเวอร์ nvidia

โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.