Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 6064/tcp, 8888/tcp

com.nvidia.build.id=126885133

NVIDIA_BUILD_ID=126885133

NVIDIA_BUILD_ID=126885133

NVIDIA_BUILD_REF=5135f7dcdc5158287853191ef360f380a0d39780

TARGETARCH=amd64 TENSORFLOW_VERSION=2.17.0 NVIDIA_TENSORFLOW_VERSION=24.12-tf2 PYVER=3.12 L4T=0 BAZEL_VERSION=6.5.0 TFAPI=2 BAZEL_CACHE= ln -sf ${_CUDA_COMPAT_PATH}/lib.real ${_CUDA_COMPAT_PATH}/lib &&
  echo ${_CUDA_COMPAT_PATH}/lib > /etc/ld.so.conf.d/00-cuda-compat.conf &&
  ldconfig &&
  rm -f ${_CUDA_COMPAT_PATH}/lib

TARGETARCH=amd64 TENSORFLOW_VERSION=2.17.0 NVIDIA_TENSORFLOW_VERSION=24.12-tf2 PYVER=3.12 L4T=0 BAZEL_VERSION=6.5.0 TFAPI=2 BAZEL_CACHE= pip install --no-cache-dir git+https://github.com/kaixih/tf_op_graph.git@v0.1

TARGETARCH=amd64 TENSORFLOW_VERSION=2.17.0 NVIDIA_TENSORFLOW_VERSION=24.12-tf2 PYVER=3.12 L4T=0 BAZEL_VERSION=6.5.0 TFAPI=2 BAZEL_CACHE= pip install pydot &&
  apt-get update &&
  apt-get install -y graphviz

TARGETARCH=amd64 TENSORFLOW_VERSION=2.17.0 NVIDIA_TENSORFLOW_VERSION=24.12-tf2 PYVER=3.12 L4T=0 BAZEL_VERSION=6.5.0 TFAPI=2 BAZEL_CACHE= if [ "${L4T}" = "1" ]; then echo "Not installing Polygraphy in iGPU container until version is specified."; else pip install --no-cache-dir --extra-index-url https://pypi.ngc.nvidia.com --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/sw-tensorrt-pypi/simple polygraphy==${POLYGRAPHY_VERSION}; fi

PATH=/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/amazon/efa/bin:/opt/tensorrt/bin