UV_VERSION=0.11.18 PYTHON_VERSION=3.13.13 MAX_JOBS= NVTE_BUILD_THREADS_PER_JOB= BUILD_CUSTOM_VLLM= BUILD_CUSTOM_VLLM_URL= BUILD_CUSTOM_VLLM_REF= BUILD_CUSTOM_VLLM_PRECOMPILED_WHEEL_LOCATION= BUILD_CUSTOM_FLASHINFER= BUILD_CUSTOM_FLASHINFER_URL= BUILD_CUSTOM_FLASHINFER_REF= SKIP_VLLM_BUILD= SKIP_SGLANG_BUILD= BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 CUSTOM_SETUP_FNAME=install_apptainer.sh NEMO_GYM_PREFETCH_CONFIGS= NEMO_RL_COMMIT=19244a0949fb363326f71675223da75e09d14866 NVIDIA_BUILD_ID=374252294 NVIDIA_BUILD_REF=f5ef697d469dc3650f03d54af6ba8b9b8e355d96 RC_DATE=v0.7.0 TARGETARCH=arm64 NEMO_GYM_CUDA=cu130 NEMO_GYM_VLLM_VERSION=0.20.0 echo "This distribution includes open source which is archived at the following URL: https://opensource.nvidia.com/oss/teams/nvidia/nemo-rl/${RC_DATE}:linux-${TARGETARCH}/index.html" > NOTICES.txt &&
  echo "For further inquiries or assistance, contact us at oss-requests@nvidia.com" >> NOTICES.txt

UV_VERSION=0.11.18 PYTHON_VERSION=3.13.13 MAX_JOBS= NVTE_BUILD_THREADS_PER_JOB= BUILD_CUSTOM_VLLM= BUILD_CUSTOM_VLLM_URL= BUILD_CUSTOM_VLLM_REF= BUILD_CUSTOM_VLLM_PRECOMPILED_WHEEL_LOCATION= BUILD_CUSTOM_FLASHINFER= BUILD_CUSTOM_FLASHINFER_URL= BUILD_CUSTOM_FLASHINFER_REF= SKIP_VLLM_BUILD= SKIP_SGLANG_BUILD= BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 CUSTOM_SETUP_FNAME=install_apptainer.sh NEMO_GYM_PREFETCH_CONFIGS= NEMO_RL_COMMIT=19244a0949fb363326f71675223da75e09d14866 NVIDIA_BUILD_ID=374252294 NVIDIA_BUILD_REF=f5ef697d469dc3650f03d54af6ba8b9b8e355d96 RC_DATE=v0.7.0 TARGETARCH=arm64 NEMO_GYM_CUDA=cu130 NEMO_GYM_VLLM_VERSION=0.20.0 python tools/generate_fingerprint.py > /opt/nemo_rl_container_fingerprint

RUN |23 UV_VERSION=0.11.18 PYTHON_VERSION=3.13.13 MAX_JOBS= NVTE_BUILD_THREADS_PER_JOB= BUILD_CUSTOM_VLLM= BUILD_CUSTOM_VLLM_URL= BUILD_CUSTOM_VLLM_REF= BUILD_CUSTOM_VLLM_PRECOMPILED_WHEEL_LOCATION= BUILD_CUSTOM_FLASHINFER= BUILD_CUSTOM_FLASHINFER_URL= BUILD_CUSTOM_FLASHINFER_REF= SKIP_VLLM_BUILD= SKIP_SGLANG_BUILD= BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 CUSTOM_SETUP_FNAME=install_apptainer.sh NEMO_GYM_PREFETCH_CONFIGS= NEMO_RL_COMMIT=19244a0949fb363326f71675223da75e09d14866 NVIDIA_BUILD_ID=374252294 NVIDIA_BUILD_REF=f5ef697d469dc3650f03d54af6ba8b9b8e355d96 RC_DATE=v0.7.0 TARGETARCH=arm64 NEMO_GYM_CUDA=cu130 NEMO_GYM_VLLM_VERSION=0.20.0 /bin/sh -c <<"EOF" bash -exu if [[ -n "${NEMO_GYM_PREFETCH_CONFIGS:-}" ]]; then case "${TARGETARCH:-amd64}" in arm64) WHEEL_ARCH=aarch64 ;; *) WHEEL_ARCH=x86_64 ;; esac GYM_VLLM_OVERRIDE=/opt/nemo-rl/.gym-vllm-override.txt printf 'vllm @ https://github.com/vllm-project/vllm/releases/download/v%s/vllm-%s-cp38-abi3-manylinux_2_35_%s.whl\n' \
"${NEMO_GYM_VLLM_VERSION}" "${NEMO_GYM_VLLM_VERSION}" "${WHEEL_ARCH}" > "${GYM_VLLM_OVERRIDE}" UV_TORCH_BACKEND="${NEMO_GYM_CUDA}" \
UV_OVERRIDE="${GYM_VLLM_OVERRIDE}" \
UV_LINK_MODE=symlink uv run python examples/nemo_gym/prefetch_venvs.py $NEMO_GYM_PREFETCH_CONFIGS rm -f "${GYM_VLLM_OVERRIDE}" fi EOF # buildkit

NEMO_GYM_VLLM_VERSION=0.20.0

NEMO_GYM_CUDA=cu130

RUN |21 UV_VERSION=0.11.18 PYTHON_VERSION=3.13.13 MAX_JOBS= NVTE_BUILD_THREADS_PER_JOB= BUILD_CUSTOM_VLLM= BUILD_CUSTOM_VLLM_URL= BUILD_CUSTOM_VLLM_REF= BUILD_CUSTOM_VLLM_PRECOMPILED_WHEEL_LOCATION= BUILD_CUSTOM_FLASHINFER= BUILD_CUSTOM_FLASHINFER_URL= BUILD_CUSTOM_FLASHINFER_REF= SKIP_VLLM_BUILD= SKIP_SGLANG_BUILD= BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 CUSTOM_SETUP_FNAME=install_apptainer.sh NEMO_GYM_PREFETCH_CONFIGS= NEMO_RL_COMMIT=19244a0949fb363326f71675223da75e09d14866 NVIDIA_BUILD_ID=374252294 NVIDIA_BUILD_REF=f5ef697d469dc3650f03d54af6ba8b9b8e355d96 RC_DATE=v0.7.0 TARGETARCH=arm64 /bin/sh -c <<"EOF" bash -exu NEGATIVE_FILTERS="" if [[ -n "${SKIP_VLLM_BUILD:-}" ]]; then NEGATIVE_FILTERS="$NEGATIVE_FILTERS vllm" fi if [[ -n "${SKIP_SGLANG_BUILD:-}" ]]; then NEGATIVE_FILTERS="$NEGATIVE_FILTERS sglang" fi if [[ -n "$NEGATIVE_FILTERS" ]]; then UV_LINK_MODE=symlink uv run nemo_rl/utils/prefetch_venvs.py --negative-filters $NEGATIVE_FILTERS else UV_LINK_MODE=symlink uv run nemo_rl/utils/prefetch_venvs.py fi EOF # buildkit

UV_VERSION=0.11.18 PYTHON_VERSION=3.13.13 MAX_JOBS= NVTE_BUILD_THREADS_PER_JOB= BUILD_CUSTOM_VLLM= BUILD_CUSTOM_VLLM_URL= BUILD_CUSTOM_VLLM_REF= BUILD_CUSTOM_VLLM_PRECOMPILED_WHEEL_LOCATION= BUILD_CUSTOM_FLASHINFER= BUILD_CUSTOM_FLASHINFER_URL= BUILD_CUSTOM_FLASHINFER_REF= SKIP_VLLM_BUILD= SKIP_SGLANG_BUILD= BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 CUSTOM_SETUP_FNAME=install_apptainer.sh NEMO_GYM_PREFETCH_CONFIGS= NEMO_RL_COMMIT=19244a0949fb363326f71675223da75e09d14866 NVIDIA_BUILD_ID=374252294 NVIDIA_BUILD_REF=f5ef697d469dc3650f03d54af6ba8b9b8e355d96 RC_DATE=v0.7.0 TARGETARCH=arm64 git rev-parse --is-shallow-repository | grep -q true &&
  git fetch --unshallow ||
  true

. /opt/nemo-rl

LD_LIBRARY_PATH=/opt/amazon/ofi-nccl/lib:/opt/amazon/efa/lib:/opt/nemo_rl_venv/lib/python3.13/site-packages/nvidia/cudnn/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64

UV_VERSION=0.11.18 PYTHON_VERSION=3.13.13 MAX_JOBS= NVTE_BUILD_THREADS_PER_JOB= BUILD_CUSTOM_VLLM= BUILD_CUSTOM_VLLM_URL= BUILD_CUSTOM_VLLM_REF= BUILD_CUSTOM_VLLM_PRECOMPILED_WHEEL_LOCATION= BUILD_CUSTOM_FLASHINFER= BUILD_CUSTOM_FLASHINFER_URL= BUILD_CUSTOM_FLASHINFER_REF= SKIP_VLLM_BUILD= SKIP_SGLANG_BUILD= BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:26.03-cuda13.2-devel-ubuntu24.04 CUSTOM_SETUP_FNAME=install_apptainer.sh NEMO_GYM_PREFETCH_CONFIGS= NEMO_RL_COMMIT=19244a0949fb363326f71675223da75e09d14866 NVIDIA_BUILD_ID=374252294 NVIDIA_BUILD_REF=f5ef697d469dc3650f03d54af6ba8b9b8e355d96 RC_DATE=v0.7.0 TARGETARCH=arm64 printf '%s\n' /opt/amazon/ofi-nccl/lib /opt/amazon/efa/lib > /etc/ld.so.conf.d/aws-efa.conf