Docker 登录英伟达 Nvidia NGC 下载容器镜像教程(以配置 CUDA 为例)

Nvidia NGC 容器

NVIDIA GPU 云计算平台(NVIDIA NGC)。该平台是一个集成式平台,专门为人工智能、机器学习和深度学习应用而设计。NGC 提供了各种优化的深度学习框架、预训练模型、AI 工具和容器,以协助开发人员快速开发和部署深度学习应用程序。

NGC 容器是基于 Docker 容器构建的深度学习应用程序,其中包含了经过优化的深度学习框架和库、各种 AI 工具和预训练模型,是 Docker Hub 的有效补充。

注:NGC 容器镜像是需要登录才能下载的,而不像 Docker Hub 可以随便下载。

教程

下面是登录 NGC 的教程。

1. 生成 NGC 登录 API

打开 NGC 容器页面,选择右上角的“Setup”:

进入 NGC Catalog 页面
进入 NGC Catalog 页面

点击“Get API Key”:

获取 Nvidia NGC API Key
获取 Nvidia NGC API Key

点击右上角的“Generate API Key”:

开始生成 Nvidia NGC API Key
开始生成 Nvidia NGC API Key

复制并保密保存生成的 API Key:

复制生成的 Nvidia NGC API Key
复制生成的 Nvidia NGC API Key

2. 用 Docker 登录 NGC nvcr.io

通过 docker login 指令登录(有时需要加 root):

docker login nvcr.io

Username 填入:$oauthtoken

这里的 $oauthtoken 就是纯粹的字符串 $oauthtoken,不是你的用户名/邮箱之类的。所有人都是 $oauthtoken

Password 填入:你刚才复制的生成的 API Key。

回车即可。

成功登录到 nvcr.io
成功登录到 nvcr.io

3. 开始下载 NGC 容器

这里就和 Docker Hub 别无二致了,直接 docker pull 即可:

# 以配置 CUDA 12 环境为例(CUDA 12.1.0)
docker pull nvcr.io/nvidia/cuda:12.1.0-devel-ubuntu22.04
拉取 CUDA 12.1 镜像
拉取 CUDA 12.1 镜像

发表评论