前言
NVIDIA Container Toolkit 可用于各种 Linux 发行版并支持不同的容器引擎。在开始之前,请确保您已经为您的 Linux 发行版安装了 NVIDIA 驱动程序。安装驱动程序的推荐方法是使用分发包管理器,但也可以使用其他安装程序机制(例如,通过.run从 NVIDIA Driver Downloads下载安装程序)
一、基本概念
Nvidia Container Toolkit,该工具使Docker的容器能与主机的Nvidia显卡进行interact。自 NVIDIAContainer Toolkit 1.7.0起,Ubuntu 18.04、Ubuntu 20.04 和 Ubuntu 22.04发行版包含对 Jetson 平台的支持。这意味着为这些发行版提供的安装说明有望在 Jetson 设备上运行。nvidia-docker2>= 2.8.0
二、操作步骤
1.添加源
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 \
&& curl -s -L https://nvidia.github.io/nvidia-container-runtime/experimental/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
2.安装重启
sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker
运行nvidia/cuda:10.0-base,如果本地没有会从网上拉取nvidia/cuda:10.0-base镜像。
sudo docker run --rm --gpus all nvidia/cuda:10.0-base nvidia-smi
表示成功。文章来源:https://www.toymoban.com/news/detail-619327.html
总结
本文介绍Ubuntu安装Nvidia Container Toolkit的流程。文章来源地址https://www.toymoban.com/news/detail-619327.html
到了这里,关于Ubuntu安装Nvidia Container Toolkit的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!