1000:1000

["-m" "nmp.customization_common.tasks.file_io" "--help"]

/opt/venv/bin/python

PATH=/opt/venv/bin:/opt/venv/bin:/root/.local/bin:/bin:/usr/local/lib/python3.12/dist-packages/torch_tensorrt/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/mpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/amazon/efa/bin:/opt/tensorrt/bin

USERNAME=ubuntu USER_UID=1000 USER_GID=1000 TRANSFORMERS_VERSION=5.8.1 apt-get update &&
  DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends --only-upgrade openssl libssl3t64 &&
  rm -rf /var/lib/apt/lists/* &&
  rm -rf /usr/local/lib/python3.12/dist-packages/aiohttp /usr/local/lib/python3.12/dist-packages/aiohttp-*.dist-info /usr/local/lib/python3.12/dist-packages/black /usr/local/lib/python3.12/dist-packages/_black* /usr/local/lib/python3.12/dist-packages/black-*.dist-info /usr/local/lib/python3.12/dist-packages/blackd /usr/local/lib/python3.12/dist-packages/tornado /usr/local/lib/python3.12/dist-packages/tornado-*.dist-info /usr/local/lib/python3.12/dist-packages/onnx /usr/local/lib/python3.12/dist-packages/onnx-*.dist-info /usr/local/lib/python3.12/dist-packages/urllib3 /usr/local/lib/python3.12/dist-packages/urllib3-*.dist-info /usr/local/lib/python3.12/dist-packages/grpc /usr/local/lib/python3.12/dist-packages/grpcio /usr/local/lib/python3.12/dist-packages/grpcio-*.dist-info /usr/local/lib/python3.12/dist-packages/wandb /usr/local/lib/python3.12/dist-packages/wandb-*.dist-info /usr/local/lib/python3.12/dist-packages/mlflow /usr/local/lib/python3.12/dist-packages/mlflow-*.dist-info /usr/local/lib/python3.12/dist-packages/mlflow_skinny-*.dist-info /usr/local/lib/python3.12/dist-packages/mistune /usr/local/lib/python3.12/dist-packages/mistune-*.dist-info /usr/local/lib/python3.12/dist-packages/PIL /usr/local/lib/python3.12/dist-packages/pillow.libs /usr/local/lib/python3.12/dist-packages/pillow-*.dist-info /usr/local/lib/python3.12/dist-packages/soupsieve /usr/local/lib/python3.12/dist-packages/soupsieve-*.dist-info /usr/local/lib/python3.12/dist-packages/jupyter_server /usr/local/lib/python3.12/dist-packages/jupyter_server-*.dist-info /usr/local/cuda/NsightSystems-cli-* /usr/local/bin/nsys /usr/local/cuda/bin/nsys

USERNAME=ubuntu USER_UID=1000 USER_GID=1000 TRANSFORMERS_VERSION=5.8.1 uv pip install --python ${VIRTUAL_ENV}/bin/python --no-cache --overrides /app/docker/customizer/preserve_base_torch.txt "aiohttp>=3.14.1,<4" "grpcio>=1.81.1,<2" "jupyter-server>=2.20.0,<3" "mistune>=3.3.3,<4" "pillow>=12.3.0,<13" "soupsieve>=2.8.4,<3" "tornado>=6.5.7,<7" "urllib3>=2.7.0,<3" "wandb>=0.28.1,<1"

USERNAME=ubuntu USER_UID=1000 USER_GID=1000 TRANSFORMERS_VERSION=5.8.1 uv pip install --python ${VIRTUAL_ENV}/bin/python --no-cache --overrides /app/docker/customizer/no_override_requirements.txt -e /app/sdk/python/nemo-platform -e /app/packages/nemo_platform_plugin -e /app/packages/nmp_common -e /app/packages/nmp_customization_common -e /app/services/core/models

USERNAME=ubuntu USER_UID=1000 USER_GID=1000 TRANSFORMERS_VERSION=5.8.1 uv pip install --python ${VIRTUAL_ENV}/bin/python --no-cache-dir --no-deps /tmp/causal-conv1d-wheel-src/wheels/cu13.2/causal_conv1d-*cp312*.whl /tmp/mamba-ssm-wheel-src/wheels/cu13.2/mamba_ssm-2.3.0-cp312*.whl

USERNAME=ubuntu USER_UID=1000 USER_GID=1000 TRANSFORMERS_VERSION=5.8.1 uv pip install --python ${VIRTUAL_ENV}/bin/python --no-cache --overrides /app/docker/customizer/preserve_base_torch.txt "transformers==${TRANSFORMERS_VERSION}" "accelerate>=1.0.0"

TRANSFORMERS_VERSION=5.8.1