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

NVIDIA_BUILD_REF

com.nvidia.build.id=26012104

NVIDIA_BUILD_ID=26012104

NVIDIA_BUILD_ID

TENSORFLOW_VERSION=1.15.5 NVIDIA_TENSORFLOW_VERSION=21.08-tf1 PYVER=3.8 BAZEL_VERSION=0.24.1 TFAPI=1 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=1.15.5 NVIDIA_TENSORFLOW_VERSION=21.08-tf1 PYVER=3.8 BAZEL_VERSION=0.24.1 TFAPI=1 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=1.15.5 NVIDIA_TENSORFLOW_VERSION=21.08-tf1 PYVER=3.8 BAZEL_VERSION=0.24.1 TFAPI=1 BAZEL_CACHE=--bazel-cache NVTX_PLUGINS_VERSION=0.1.8 URL=$(VERIFY=1 /nvidia/build-scripts/installTRT.sh 2>/dev/null | sed -n "s/^.*\(http.*\)deb.*$/\1/p")tar &&
  FILE=$(wget -O - $URL 2>/dev/null | sed -n 's/^.*href="\(TensorRT[^"]*\)".*$/\1/p' | grep -v internal) &&
  wget --quiet $URL/$FILE -O - | tar -xz &&
  PY=$(python -c 'import sys; print(str(sys.version_info[0])+str(sys.version_info[1]))') &&
  pip install TensorRT-*/python/tensorrt-*-cp$PY*.whl &&
  pip install TensorRT-*/graphsurgeon/graphsurgeon-*.whl &&
  pip install TensorRT-*/uff/uff-*.whl &&
  mv /usr/src/tensorrt /opt &&
  ln -s /opt/tensorrt /usr/src/tensorrt &&
  rm -r TensorRT-* &&
  UFF_PATH=$(pip show uff | sed -n 's/Location: \(.*\)$/\1/p')/uff &&
  chmod +x ${UFF_PATH}/bin/convert_to_uff.py &&
  ln -sf ${UFF_PATH}/bin/convert_to_uff.py /usr/local/bin/convert-to-uff

TENSORFLOW_VERSION=1.15.5 NVIDIA_TENSORFLOW_VERSION=21.08-tf1 PYVER=3.8 BAZEL_VERSION=0.24.1 TFAPI=1 BAZEL_CACHE=--bazel-cache NVTX_PLUGINS_VERSION=0.1.8 cd /workspace/nvidia-examples/OpenSeq2Seq &&
  patch -p0 < openseq2seq.patch &&
  ./scripts/install_kenlm.sh &&
  cd /opt/tensorflow/tensorflow-source &&
  ln -s /workspace/nvidia-examples/OpenSeq2Seq/ctc_decoder_with_lm ./tensorflow/core/user_ops/ &&
  ../nvbuild.sh --configonly --python$PYVER &&
  bazel build $(cat /opt/tensorflow/nvbuildopts) //tensorflow/core/user_ops/ctc_decoder_with_lm:libctc_decoder_with_kenlm.so //tensorflow/core/user_ops/ctc_decoder_with_lm:generate_trie &&
  cp bazel-bin/tensorflow/core/user_ops/ctc_decoder_with_lm/*.so tensorflow/core/user_ops/ctc_decoder_with_lm/ &&
  cp bazel-bin/tensorflow/core/user_ops/ctc_decoder_with_lm/generate_trie tensorflow/core/user_ops/ctc_decoder_with_lm/ &&
  bazel clean --expunge &&
  rm .tf_configure.bazelrc /usr/local/cuda/lib64/stubs/libcuda.so.1 &&
  rm -rf ${HOME}/.cache/bazel /tmp/*