当前位置：首页 > article >正文

NVIDIA PyTorch Docker 镜像安装

article 2025/1/16 14:40:10

nvcr.io/nvidia/pytorch:24.12-py3-igpu 是一个 NVIDIA 提供的 PyTorch Docker 镜像，其中包含了 PyTorch 以及与 NVIDIA GPU 相关的库，24.12 表示这个镜像的版本号， py3 表示python3版本，igpu表示集显。

步骤：

确保你已安装 Docker:
- 如果你的系统上还没有安装 Docker，请先安装 Docker Desktop（适用于 macOS 和 Windows）或者 Docker Engine（适用于 Linux）。
- 你可以从 Docker 官网下载并安装：https://www.docker.com/get-started
- 安装完成后，运行 docker --version 确认 Docker 是否安装成功。
拉取 Docker 镜像:
- 打开你的终端或命令提示符。
- 使用 docker pull 命令拉取镜像：
```
docker pull nvcr.io/nvidia/pytorch:24.12-py3-igpu
```
  - docker pull: 是 Docker 的拉取镜像命令。
  - nvcr.io/nvidia/pytorch:24.12-py3-igpu: 是要拉取的镜像的名称，这个镜像存储在 nvcr.io 的镜像仓库中。
- Docker 会自动从镜像仓库下载镜像。这个过程可能会需要一些时间，取决于你的网络速度和镜像大小。
运行 Docker 镜像:
- 拉取完成后，你可以使用 docker run 命令运行镜像：
```
docker run --gpus all -it nvcr.io/nvidia/pytorch:24.12-py3-igpu /bin/bash
```
  - docker run: 是 Docker 的运行容器命令。
  - --gpus all: 指定使用所有的GPU资源，如果只想使用部分，可以更换为--gpus device=<gpu_id>
  - -it: 以交互式的方式运行容器，并且分配一个伪终端。
  - nvcr.io/nvidia/pytorch:24.12-py3-igpu: 指定要运行的镜像名称。
  - /bin/bash: 在容器中启动 bash shell。你可以使用其他命令来覆盖默认启动命令。
- 如果想要使用指定的端口，需要增加 -p 参数：
```
docker run -p 8888:8888 --gpus all -it nvcr.io/nvidia/pytorch:24.12-py3-igpu /bin/bash
```
  其中 8888:8888 表示将容器的8888端口映射到宿主机的8888端口，这样就能从宿主机访问容器中的服务。
- 运行此命令后，你将进入容器的 bash shell。你可以在这里运行 PyTorch 代码。
- --gpus all 参数指定使用所有可用的 GPU，如果你想使用特定的 GPU，可以例如使用 --gpus '"device=0,2"' 指定使用 0 和 2 号 GPU。
- 如果需要挂载宿主机的目录到容器中，可以使用 -v 参数:
```
docker run -v /path/in/host:/path/in/container --gpus all -it nvcr.io/nvidia/pytorch:24.12-py3-igpu /bin/bash
```
  其中 /path/in/host 是宿主机的文件路径， /path/in/container 是容器中的路径。
退出 Docker 容器:
- 在容器的 bash shell 中，输入 exit 并按回车键即可退出容器。