博客

在centos docker中安装nvidia驱动

  • Post category:other

在CentOS Docker中安装NVIDIA驱动的完整攻略

在CentOS Docker中安装NVIDIA驱动可以让我们在Docker容器中使用GPU加速。本文为您提供在CentOS Docker中安装NVIDIA驱动的完整攻略，包括以下内容：

  1. 安装NVIDIA驱动
  2. 配置Docker以使用NVIDIA驱动
  3. 示例说明

安装NVIDIA驱动

首先，我们需要安装NVIDIA驱动。可以使用以下命令：

sudo yum install -y kernel-devel-$(uname -r) kernel-headers-$(uname -r) gcc make
sudo yum install -y epel-release
sudo yum install -y dkms
sudo yum install -y wget
wget http://us.download.nvidia.com/XFree86/Linux-x86_64/460.39/NVIDIA-Linux-x86_64-460.39.run
sudo sh NVIDIA-Linux-x86_64-460.39.run

这将安装NVIDIA驱动。

配置Docker以使用NVIDIA驱动

接下来，我们需要配置Docker以使用NVIDIA驱动。可以使用以下命令：

sudo yum install -y nvidia-container-toolkit
sudo systemctl restart docker

这将安装nvidia-container-toolkit并重启Docker服务。

示例说明

以下是两个示例：

示例1：在Docker容器中使用NVIDIA驱动

  1. 启动一个Docker容器：
sudo docker run --gpus all -it centos:7 /bin/bash

这将启动一个CentOS 7的Docker容器，并使用所有可用的GPU。

  1. 在Docker容器中安装CUDA：
wget https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-repo-rhel7-11.4.0-1.x86_64.rpm
sudo rpm -i cuda-repo-rhel7-11.4.0-1.x86_64.rpm
sudo yum clean all
sudo yum install -y cuda

这将在Docker容器中安装CUDA。

示例2：在Docker容器中运行TensorFlow

  1. 启动一个Docker容器：
sudo docker run --gpus all -it tensorflow/tensorflow:latest-gpu /bin/bash

这将启动一个TensorFlow的Docker容器，并使用所有可用的GPU。

  1. 在Docker容器中运行TensorFlow：
python
import tensorflow as tf
print(tf.config.list_physical_devices('GPU'))

这将在Docker容器中运行TensorFlow，并输出可用的GPU设备。

Tags: