Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/cosmos_rl; ExposedPorts 8000/tcp

RUN_CLI=0

FLASK_APP=nvidia_tao_core.microservices.app

8000/tcp

TAO_TELEMETRY_SERVER=https://api.tao.ngc.nvidia.com

NVIDIA_TAO_TOOLKIT_VERSION=6.25.7-PyTorch

TAO_TOOLKIT_VERSION=6.25.7

NVIDIA_PRODUCT_NAME=TAO Toolkit

GDRCOPY_VERSION=v2.4.4 EFA_INSTALLER_VERSION=1.42.0 AWS_OFI_NCCL_VERSION=v1.16.0 NCCL_VERSION=2.25.1-1 PYTHON_VERSION=3.12 pip uninstall -y ray

GDRCOPY_VERSION=v2.4.4 EFA_INSTALLER_VERSION=1.42.0 AWS_OFI_NCCL_VERSION=v1.16.0 NCCL_VERSION=2.25.1-1 PYTHON_VERSION=3.12 . /opt/venv/cosmos_rl/bin/activate &&
  cd tao-core &&
  bash release/python/build_wheel.sh &&
  find dist/ -name "nvidia_tao_core*.whl" -type f | xargs -n 1 pip install &&
  cp nvidia_tao_core/microservices/nginx.conf /etc/nginx/ &&
  cd .. &&
  rm -rf tao-core