在CentOS上安装nvidia-container-runtime,首先需要确保你的系统已经安装了NVIDIA的驱动和docker。以下是安装步骤:
确保Docker已安装:
sudo yum install -y yum-utils
sudo yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
sudo yum install -y docker-ce docker-ce-cli containerd.io
sudo systemctl start docker
sudo systemctl enable docker
添加NVIDIA Docker repository:
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo rpm --import
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo
安装nvidia-container-runtime:
sudo yum install -y nvidia-container-runtime
重启Docker服务:
sudo systemctl restart docker
确保你的系统有GPU并且安装了NVIDIA驱动。如果驱动没有安装,你可以使用如下命令安装:
sudo yum install nvidia-driver nvidia-settings
sudo reboot
安装完成后,你可以通过运行一个测试的NVIDIA容器来验证nvidia-container-runtime是否正确安装和配置:
docker run --gpus all nvidia/cuda:10.0-base nvidia-smi
这个命令会启动一个NVIDIA CUDA镜像,并且运行nvidia-smi命令来显示GPU信息,如果安装正确,你将看到你的GPU列表和驱动版本信息。