在Docker容器中使用Nvidia GPU

Installation

Installing with Apt

  1. Configure the production repository:$ 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/stable/deb/nvidia-container-toolkit.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 Optionally, configure the repository to use experimental packages:$ sed -i -e '/experimental/ s/^#//g' /etc/apt/sources.list.d/nvidia-container-toolkit.list
  2. Update the packages list from the repository:$ sudo apt-get update
  3. Install the NVIDIA Container Toolkit packages:$ sudo apt-get install -y nvidia-container-toolkit

Installing with Yum or Dnf

  1. Configure the production repository:$ curl -s -L https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo \ sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo Optionally, configure the repository to use experimental packages:$ sudo yum-config-manager --enable nvidia-container-toolkit-experimental
  2. Install the NVIDIA Container Toolkit packages:$ sudo yum install -y nvidia-container-toolkit

Installing with Zypper

  1. Configure the production repository:$ sudo zypper ar https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo Optionally, configure the repository to use experimental packages:$ sudo zypper modifyrepo --enable nvidia-container-toolkit-experimental
  2. Install the NVIDIA Container Toolkit packages:$ sudo zypper --gpg-auto-import-keys install -y nvidia-container-toolkit

Configuration

Prerequisites

  • You installed a supported container engine (Docker, Containerd, CRI-O, Podman).
  • You installed the NVIDIA Container Toolkit.

Configuring Docker

  1. Configure the container runtime by using the nvidia-ctk command:$ sudo nvidia-ctk runtime configure --runtime=docker The nvidia-ctk command modifies the /etc/docker/daemon.json file on the host. The file is updated so that Docker can use the NVIDIA Container Runtime.
  2. Restart the Docker daemon:$ sudo systemctl restart docker

Rootless mode

To configure the container runtime for Docker running in Rootless mode, follow these steps:

  1. Configure the container runtime by using the nvidia-ctk command:$ nvidia-ctk runtime configure --runtime=docker --config=$HOME/.config/docker/daemon.json
  2. Restart the Rootless Docker daemon:$ systemctl --user restart docker
  3. Configure /etc/nvidia-container-runtime/config.toml by using the sudo nvidia-ctk command:$ sudo nvidia-ctk config --set nvidia-container-cli.no-cgroups --in-place

Configuring containerd (for Kubernetes)

  1. Configure the container runtime by using the nvidia-ctk command:$ sudo nvidia-ctk runtime configure --runtime=containerd The nvidia-ctk command modifies the /etc/containerd/config.toml file on the host. The file is updated so that containerd can use the NVIDIA Container Runtime.
  2. Restart containerd:$ sudo systemctl restart containerd

Configuring containerd (for nerdctl)

No additional configuration is needed. You can just run nerdctl run --gpus=all, with root or without root. You do not need to run the nvidia-ctk command mentioned above for Kubernetes.

See also the nerdctl documentation.

Configuring CRI-O

  1. Configure the container runtime by using the nvidia-ctk command:$ sudo nvidia-ctk runtime configure --runtime=crio The nvidia-ctk command modifies the /etc/crio/crio.conf file on the host. The file is updated so that CRI-O can use the NVIDIA Container Runtime.
  2. Restart the CRI-O daemon:$ sudo systemctl restart crio

Configuring Podman

For Podman, NVIDIA recommends using CDI for accessing NVIDIA devices in containers.

Next Steps

  • Install an NVIDIA GPU Driver if you do not already have one installed. You can install a driver by using the package manager for your distribution, but other installation methods, such as downloading a .run file intaller, are available. Refer to the NVIDIA Driver Installation Quickstart Guide for more information.
  • Running a Sample Workload
暂无评论

发送评论 编辑评论


				
|´・ω・)ノ
ヾ(≧∇≦*)ゝ
(☆ω☆)
(╯‵□′)╯︵┴─┴
 ̄﹃ ̄
(/ω\)
∠( ᐛ 」∠)_
(๑•̀ㅁ•́ฅ)
→_→
୧(๑•̀⌄•́๑)૭
٩(ˊᗜˋ*)و
(ノ°ο°)ノ
(´இ皿இ`)
⌇●﹏●⌇
(ฅ´ω`ฅ)
(╯°A°)╯︵○○○
φ( ̄∇ ̄o)
ヾ(´・ ・`。)ノ"
( ง ᵒ̌皿ᵒ̌)ง⁼³₌₃
(ó﹏ò。)
Σ(っ °Д °;)っ
( ,,´・ω・)ノ"(´っω・`。)
╮(╯▽╰)╭
o(*////▽////*)q
>﹏<
( ๑´•ω•) "(ㆆᴗㆆ)
😂
😀
😅
😊
🙂
🙃
😌
😍
😘
😜
😝
😏
😒
🙄
😳
😡
😔
😫
😱
😭
💩
👻
🙌
🖕
👍
👫
👬
👭
🌚
🌝
🙈
💊
😶
🙏
🍦
🍉
😣
Source: github.com/k4yt3x/flowerhd
颜文字
Emoji
小恐龙
花!
上一篇
下一篇