Ubuntu 22.04 LTS中安装Docker,并支持显卡

发布于:2024-02-28 ⋅ 阅读:(991) ⋅ 点赞:(0)

步骤如下:

# 第1步:更新软件存储库
sudo apt-get update

# 第2步:卸载旧版本的Docker
sudo apt-get remove docker docker-engine docker.io

# 第3步:在 Ubuntu 18.04 上安装Docker
sudo apt install docker.io

# 第4步:安装 nvidia-docker
# 参考: https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
# Setup the package repository and the GPG key:
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
      && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
      && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
            sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
            sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

# 更新源和安装 nvidia-container-toolkit
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit

# 配置Docker守护进程以识别NVIDIA容器运行时
sudo nvidia-ctk runtime configure --runtime=docker

# 重启 Docker
sudo systemctl restart docker

# 测试
# pull nvidia/cuda:12.3.1-base-ubuntu22.04
# run --rm --runtime=nvidia --gpus all nvidia/cuda:12.3.1-base-ubuntu22.04 nvidia-smi

正常的话,测试结果显示:
image.png

本文含有隐藏内容,请 开通VIP 后查看

网站公告

今日签到

点亮在社区的每一天
去签到