NVIDIA
NVIDIA
PyTorch
Container
NVIDIA
NVIDIA
PyTorch

PyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels.

LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.ref=d8da1b277af0d8de8866e4b1ee1ce709a890706d
11/26/2020 11:25 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_REF
11/26/2020 11:25 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=17950526
11/26/2020 11:25 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=17950526
11/26/2020 11:25 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID
11/26/2020 11:25 AM UTC
sha256:a5ff18c9283bbde404142feda3b8eb99007923402135d6db66d2ecfdae616135RUN
NVIDIA_PYTORCH_VERSION=20.12 PYTORCH_BUILD_VERSION=1.8.0a0+1606899 PYVER=3.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
11/26/2020 11:25 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
PATH=/opt/conda/bin:/opt/cmake-3.14.6-Linux-x86_64/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
11/26/2020 11:25 AM UTC
sha256:2992460da2560be71c3de5e6738177ccaf9c79e609d50b2f28ec0ea6c7d9ebddRUN
NVIDIA_PYTORCH_VERSION=20.12 PYTORCH_BUILD_VERSION=1.8.0a0+1606899 PYVER=3.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 &&
  sed -i 's/from tensorflow import GraphDef/from tensorflow.python import GraphDef/' $UFF_PATH/converters/tensorflow/conversion_helpers.py &&
  chmod +x ${UFF_PATH}/bin/convert_to_uff.py &&
  ln -sf ${UFF_PATH}/bin/convert_to_uff.py /usr/local/bin/convert-to-uff
11/26/2020 11:25 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
/usr/local/bin/nvidia_entrypoint.sh
11/26/2020 11:25 AM UTC
sha256:e0466816640f8b5c21bbad873d21ff41854698c576ebb5538672eca4caac5470COPY
nvidia_entrypoint.sh /usr/local/bin
11/26/2020 11:25 AM UTC
...