如何在 CentOS 安装 GPU 驱动

云计算 2023-01-04 泡泡 手机阅读

以 CentOS 7.7，Tesla P100 GPU 为例。

1. 基础环境准备

  • 安装 lspci 命令
1

yum install -y pciutils
  • 检查 GPU 是否支持 CUDA
1
2
3

lspci | grep -i nvidia

00:09.0 3D controller: NVIDIA Corporation GP100GL [Tesla P100 PCIe 12GB] (rev a1)
  • 第二种，下载官网驱动安装
1
2
3

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo
yum install -y nvidia-container-runtime nvidia-container-toolkit nvidia-docker2
  • 添加新的 runtime