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

NVIDIA_BUILD_REF

com.nvidia.build.id=28019337

NVIDIA_BUILD_ID=28019337

NVIDIA_BUILD_ID

entrypoint.d/ /opt/nvidia/entrypoint.d/

NVIDIA_PYTORCH_VERSION=21.10 PYTORCH_BUILD_VERSION=1.10.0a0+0aef44c TARGETARCH=amd64 PYVER=3.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

NVIDIA_PYTORCH_VERSION=21.10 PYTORCH_BUILD_VERSION=1.10.0a0+0aef44c TARGETARCH=amd64 PYVER=3.8 mkdir -p /opt/dlprof &&
  mkdir -p /opt/dlprof_viewer_install &&
  cp /nvidia/tmp/pip/* /opt/dlprof_viewer_install/ &&
  cp /nvidia/opt/dlprof/bin/dlprof /opt/dlprof/ &&
  cp /nvidia/workspace/LICENSE /opt/dlprof/ &&
  cp /nvidia/workspace/README.md /opt/dlprof/ &&
  ln -sf /opt/dlprof/dlprof /usr/local/bin/dlprof &&
  cd /nvidia/opt/dlprof/bin/ &&
  pip install --no-cache-dir nvidia_dlprof_pytorch_nvtx* &&
  cd /opt/dlprof_viewer_install &&
  pip install --no-cache-dir nvidia_dlprofviewer-* &&
  rm -rf /opt/conda/lib/python3.8/site-packages/qa

NVIDIA_PYTORCH_VERSION=21.10 PYTORCH_BUILD_VERSION=1.10.0a0+0aef44c TARGETARCH=amd64 PYVER=3.8 pip --version &&
  python -c 'import sys; print(sys.platform)' &&
  pip install --no-cache-dir nvidia-pyindex &&
  pip install --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/sw-tensorrt-pypi/simple --no-cache-dir polygraphy==0.33.0 &&
  if [[ $TARGETARCH = "amd64" ]] ; then pip install --extra-index-url http://sqrl/dldata/pip-simple --trusted-host sqrl --no-cache-dir pytorch-quantization==2.1.0; fi

PATH=/opt/conda/bin:/opt/cmake-3.19.3-Linux-$(uname -m)/bin/:/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