1. 程式人生 > 其它 >【煉丹】CentOS 7安裝GPU(Tesla P100)驅動

【煉丹】CentOS 7安裝GPU(Tesla P100)驅動

技術標籤:煉丹筆記自然語言處理神經網路深度學習

下載驅動

https://www.nvidia.cn/Download/index.aspx?lang=cn

CentOS 7操作

1. 啟動方式 需要是 UEFI/EFI非安全啟動方式

2. 基礎環境準備

  1. 檢查 GPU 是否支援 CUDA
lspci | grep -i nvidia 

00:09.0 3D controller: NVIDIA Corporation GP100GL [Tesla P100 PCIe 12GB] (rev a1)
  1. 檢查系統是否支援CUDA
uname -m && cat /etc/redhat-release 
x86_64 
CentOS Linux release 7.7.1908 (
Core)
  1. 安裝系統工具包
yum update -y
yum install -y wget vim gcc
yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)

3. 安裝GPU驅動

  1. 禁用系統預設的 nouveau 驅動
    遮蔽前:
lsmod | grep nouveau 

在這裡插入圖片描述

  1. 禁用 nouveau
bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
  1. 重建 initramfs image
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak 
dracut /boot/initramfs-$(uname -r).img $(uname -r)
  1. 重啟系統
reboot
  1. 遮蔽nouveau之後
lsmod | grep nouveau 
(結果為空)
  1. 安裝驅動
chmod +x NVIDIA-Linux-x86_64-440.64.00.run 
bash ./NVIDIA-Linux-x86_64-440.64.00.run