Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 6064/tcp, 8888/tcp

NVIDIA_BUILD_REF

com.nvidia.build.id=17379986

NVIDIA_BUILD_ID=17379986

NVIDIA_BUILD_ID

TENSORFLOW_VERSION=2.3.1 NVIDIA_TENSORFLOW_VERSION=20.11-tf2 PYVER=3.6 BAZEL_VERSION=3.1.0 TFAPI=2 BAZEL_CACHE=--bazel-cache NVTX_PLUGINS_VERSION=0.1.8 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

TENSORFLOW_VERSION=2.3.1 NVIDIA_TENSORFLOW_VERSION=20.11-tf2 PYVER=3.6 BAZEL_VERSION=3.1.0 TFAPI=2 BAZEL_CACHE=--bazel-cache NVTX_PLUGINS_VERSION=0.1.8 pip install --no-cache-dir --extra-index-url https://pypi.ngc.nvidia.com --extra-index-url https://tensorrt-read-only:Tensorrt\@123@urm.nvidia.com/artifactory/api/pypi/sw-tensorrt-pypi/simple polygraphy

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/tensorrt/bin

TENSORFLOW_VERSION=2.3.1 NVIDIA_TENSORFLOW_VERSION=20.11-tf2 PYVER=3.6 BAZEL_VERSION=3.1.0 TFAPI=2 BAZEL_CACHE=--bazel-cache NVTX_PLUGINS_VERSION=0.1.8 cd /opt/tensorrt/python &&
  ./python_setup.sh

trt_python_setup.sh /opt/tensorrt/python/python_setup.sh