Skip to main content
NVIDIA
nemo-gym
Container
NVIDIA
nemo-gym

NVIDIA NeMo™ Gym is an open-source library under the NVIDIA NeMo Framework for evaluating and improving models and agents using environments.

LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
--help
10/08/2026 7:48 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
gym
10/08/2026 7:48 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4USER
root
10/08/2026 7:48 PM UTC
sha256:491f3d98fcd5c1c2e8e3f1f21f370552a79e281ae5c2c25325b4f4f205427e7eRUN
ENROOT_VERSION=3.5.0 UV_VERSION=0.11.29 PYTHON_VERSION=3.13.14 TARGETARCH=arm64 RUNTIME_UID=65532 RUNTIME_GID=65532 BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 NEMO_GYM_PREFETCH_CONFIGS= NEMO_GYM_COMMIT=1f8afc71d45fac14872d86fe300374afa2001e7a NVIDIA_BUILD_ID=476771270 NVIDIA_BUILD_REF=8c8da7764bf7afe7195a453bd1dfb80ef22ff7d1 RC_DATE=v0.7.0 NEMO_GYM_SEED_SERVER_ENVS= export HOME=/home/nemo-runtime XDG_CACHE_HOME=/opt/nemo-gym/cache/xdg &&
  test "$(id -u)" = "${RUNTIME_UID}" &&
  test -w "${HOME}" &&
  test -w "${XDG_CACHE_HOME}" &&
  test -w "${UV_CACHE_DIR}" &&
  test -w "${PRE_COMMIT_HOME}" &&
  test -w /opt/nemo-gym/cache &&
  test -w /opt/nemo-gym/results &&
  test -x /usr/local/bin/uv &&
  test -x /usr/local/bin/uvx &&
  test -x /opt/nemo_gym_venv/bin/python &&
  venv_site_packages="$(python -c 'import site; print(site.getsitepackages()[0])')" &&
  rm "${venv_site_packages}/.nonroot-uninstall-probe" &&
  touch "${HOME}/.nonroot-write-probe" "${XDG_CACHE_HOME}/.nonroot-write-probe" "${UV_CACHE_DIR}/.nonroot-write-probe" /opt/nemo-gym/.nonroot-write-probe /opt/nemo-gym/cache/.nonroot-write-probe /opt/nemo-gym/results/.nonroot-write-probe "${venv_site_packages}/.nonroot-write-probe" &&
  gym --help >/dev/null &&
  uv venv /opt/nemo-gym/cache/container-smoke &&
  /opt/nemo-gym/cache/container-smoke/bin/python -c "import sys; assert sys.version_info[:2] == (3, 13)" &&
  rm -r /opt/nemo-gym/cache/container-smoke "${HOME}/.nonroot-write-probe" "${XDG_CACHE_HOME}/.nonroot-write-probe" "${UV_CACHE_DIR}/.nonroot-write-probe" /opt/nemo-gym/.nonroot-write-probe /opt/nemo-gym/cache/.nonroot-write-probe /opt/nemo-gym/results/.nonroot-write-probe "${venv_site_packages}/.nonroot-write-probe"
10/08/2026 7:48 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4USER
65532:65532
10/08/2026 7:48 PM UTC
sha256:dc0bf53cf3d7ec803c4fa917d1d2f0f54c83e81d25bc828a0e67bad0e073b013RUN
ENROOT_VERSION=3.5.0 UV_VERSION=0.11.29 PYTHON_VERSION=3.13.14 TARGETARCH=arm64 RUNTIME_UID=65532 RUNTIME_GID=65532 BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 NEMO_GYM_PREFETCH_CONFIGS= NEMO_GYM_COMMIT=1f8afc71d45fac14872d86fe300374afa2001e7a NVIDIA_BUILD_ID=476771270 NVIDIA_BUILD_REF=8c8da7764bf7afe7195a453bd1dfb80ef22ff7d1 RC_DATE=v0.7.0 NEMO_GYM_SEED_SERVER_ENVS= venv_site_packages="$(/opt/nemo_gym_venv/bin/python -c         'import site; print(site.getsitepackages()[0])')" &&
  touch "${venv_site_packages}/.nonroot-uninstall-probe" &&
  find /opt/nemo-gym -path /opt/nemo-gym/cache -prune -o -type d -name .venv -prune -o -type d -exec chown "${RUNTIME_UID}:${RUNTIME_GID}" {} + &&
  find /opt/nemo_gym_venv -type d -exec chown "${RUNTIME_UID}:${RUNTIME_GID}" {} +
10/08/2026 7:48 PM UTC
sha256:4222b786e3a929c0cd5106341d9e12a52f1302d64d807df5c661f898d1a52661RUN
RUN |13 ENROOT_VERSION=3.5.0 UV_VERSION=0.11.29 PYTHON_VERSION=3.13.14 TARGETARCH=arm64 RUNTIME_UID=65532 RUNTIME_GID=65532 BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 NEMO_GYM_PREFETCH_CONFIGS= NEMO_GYM_COMMIT=1f8afc71d45fac14872d86fe300374afa2001e7a NVIDIA_BUILD_ID=476771270 NVIDIA_BUILD_REF=8c8da7764bf7afe7195a453bd1dfb80ef22ff7d1 RC_DATE=v0.7.0 NEMO_GYM_SEED_SERVER_ENVS= /bin/sh -c <<"EOF" bash -exu if [[ -n "${NEMO_GYM_PREFETCH_CONFIGS:-}" ]]; then gym env prefetch "+config_paths=[${NEMO_GYM_PREFETCH_CONFIGS}]" fi # ray's bundled Java worker jar ships an outdated httpcore5 with a known CVE
# (GHSA-hf6x-8p5f-cgmf). Gym never uses Ray's Java cross-language worker, and # removing the jar doesn't affect ray.init(), remote tasks, actors, or
# ray.serve (verified locally). Covers both installed venvs (site-packages)
# and uv's own cache (archive-v0): confirmed uv silently skips a missing # cached file and installs everything else fine, even fully offline, so # purging it from the cache too doesn't break installing a fresh venv later
# for a server not prefetched here.
find /opt -name ray_dist.jar -delete

# Keep ownership changes for generated caches and server venvs in the layer
# that creates them so later layers do not duplicate their contents.
chown -R "${RUNTIME_UID}:${RUNTIME_GID}" /opt/nemo-gym/cache
find /opt/nemo-gym -type d -name .venv -prune \
    -exec chown -R "${RUNTIME_UID}:${RUNTIME_GID}" {} +
EOF # buildkit
10/08/2026 7:48 PM UTC
sha256:38a7761e9a9fd3d16c86b572f4653b88b1134833525846d8261541f9aba3daccRUN
ENROOT_VERSION=3.5.0 UV_VERSION=0.11.29 PYTHON_VERSION=3.13.14 TARGETARCH=arm64 RUNTIME_UID=65532 RUNTIME_GID=65532 BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 NEMO_GYM_PREFETCH_CONFIGS= NEMO_GYM_COMMIT=1f8afc71d45fac14872d86fe300374afa2001e7a NVIDIA_BUILD_ID=476771270 NVIDIA_BUILD_REF=8c8da7764bf7afe7195a453bd1dfb80ef22ff7d1 RC_DATE=v0.7.0 NEMO_GYM_SEED_SERVER_ENVS= install -d -o "${RUNTIME_UID}" -g "${RUNTIME_GID}" "${UV_CACHE_DIR}" /opt/nemo-gym/cache/xdg
10/08/2026 7:47 PM UTC
sha256:c860de5000891a25f75fd8ad884021330ebaf60fb7d485f62961ef2d97b158dcRUN
ENROOT_VERSION=3.5.0 UV_VERSION=0.11.29 PYTHON_VERSION=3.13.14 TARGETARCH=arm64 RUNTIME_UID=65532 RUNTIME_GID=65532 BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 NEMO_GYM_PREFETCH_CONFIGS= NEMO_GYM_COMMIT=1f8afc71d45fac14872d86fe300374afa2001e7a NVIDIA_BUILD_ID=476771270 NVIDIA_BUILD_REF=8c8da7764bf7afe7195a453bd1dfb80ef22ff7d1 RC_DATE=v0.7.0 NEMO_GYM_SEED_SERVER_ENVS= git init --quiet &&
  pre-commit install-hooks &&
  chown -R "${RUNTIME_UID}:${RUNTIME_GID}" "${PRE_COMMIT_HOME}" &&
  rm -rf .git
10/08/2026 7:47 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
PRE_COMMIT_HOME=/opt/nemo-gym/cache/pre-commit
10/08/2026 7:47 PM UTC
1-10 of 127 items

Modal Content

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.