PyTorch - NVIDIA AI Workbench Default Container

NVIDIA AI Workbench provides a set of default base images, referred to as Base Environments in the application, that can be used as the starting point when creating the container for a new project.

This is a Pytorch image with PyTorch 2.4.0 and CUDA 12.5 and with JupyterLab and Tensorboard applications installed.

CHANGELOG

All notable changes to this image are documented here

[1.0.1] - 2023-10

• First draft of image, based on nvidia/pytorch:23.09-py3
• Installed git-lfs
• Upgraded Jupyterlab to 4.0.7

[1.0.2] - 2023-11

• Added sources for apt packages

[1.0.3] - 2023-12

• Re-built with updated base image

[1.0.4] - 2024-XX

• Re-built with updated base image to nvidia/pytorch:24.02-py3
• Upgraded CUDA version to 12.3
• Upgraded Jupyterlab to 4.1.2
• Disabled Jupyterlab announcement banner

[1.0.5] - 2024-08

• Re-built with updated base image to pytorch:24.07-py3
• Upgraded CUDA version to 12.5
• Upgraded Jupyterlab to 4.2.4

[1.0.6] - 2025-02

• Re-built with updated base image to pytorch:24.11-py3
• Upgraded CUDA version to 12.6.3
• Upgraded Jupyterlab to 4.2.5

Shared Responsibility Model

NVIDIA AI Workbench employs a shared responsibility model. The subset of responsibilities related to base images are:

NVIDIA Responsibilities

1. Updating base environment images with a newer tag in case a critical or high CVE is discovered.
2. Providing prompt notifications in the case of updates via
   • Notifications posted on the program DevZone site
   • Email to the accounts registered in the Early Access program on DevZone

End User and Organization Responsibilities

1. Following notifications from NVIDIA, ensuring the Default Base Images are updated to the latest versions to mitigate risks associated with vulnerabilities that might impact older images.

Open Source 3rd Party Notices

The 3rd party notices file can be found in the container at /NOTICES