Nvidia驱动

【亲测有效】Linux系统安装NVIDIA显卡驱动_linux安装nvidia显卡驱动-CSDN博客

Download the latest official NVIDIA drivers

nvidia 驱动安装

Docker+Nvidia Container Toolkit

【Docker】Docker及Nvidia Container Toolkit安装-CSDN博客

Docker绑定GPU

Runtime options with Memory, CPUs, and GPUs

docker run -it --rm --gpus '"device=0,2"' ubuntu nvidia-smi
 
Exposes the first and third GPUs.
公开第一个和第三个 GPU。

问题

有些安装 Nvidia docker2，最后也是用—gpuall 启动

有这么说的：nvidia-container toolkit 是nvidia-docker2 的高级版本，装前面一个就够了

nvidia docker, nvidia docker2, nvidia container toolkits三者的区别

安装Nvidia docker2

1. 安装NVIDIA 驱动程序
2. 安装nvidia-Container-toolkit
3. https://docs.docker.com/config/containers/resource_constraints/#install-nvidia-container-toolkit

# 安装环境
# 1.nvidia-smi
nvidia-smi
# 2.NVIDIA Container Toolkit
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \\
   && curl -s -L <https://nvidia.github.io/nvidia-docker/gpgkey> | sudo apt-key add - \\
   && curl -s -L <https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list> | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
# 更新安装
apt-get update
apt-get install -y nvidia-docker2
# 重启
sudo systemctl restart docker

# 测试
docker run -it --gpus all nvidia/cuda:11.4.0-base-ubuntu20.04 nvidia-smi

操作