NVIDIA AI Workbench provides a set of default base images, referred to as Base Environments in the application, that are used as the starting point when creating the container for a new project.
This is a PyTorch image with PyTorch 2.1.0 and CUDA 12.2 and with JupyterLab and Tensorboard applications installed that is configured for use with AI Workbench. It is based off of https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch.
AI Workbench is currently in Early Access. To access the software and start using this conatiner, first sign up here: https://developer.nvidia.com/ai-workbench-early-access
Use of this container falls under the AI Workbench EULA, which is available in DevZone after acceptance into the Early Access program.
All notable changes to this image are documented here
NVIDIA AI Workbench employs a shared responsibility model. The subset of responsibilities related to base images are: