Skip to main content
NVIDIA
PyTorch
Container
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=706dbb20f62dd16a38ecdf747ebc479c614b8722
    06/17/2021 11:02 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF
    06/17/2021 11:02 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=24129275
    06/17/2021 11:02 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=24129275
    06/17/2021 11:02 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID
    06/17/2021 11:02 PM UTC
    sha256:85f9980f841b99969cc28b4da6b4961bc046d354202f456e8e811e008a9eb7e6RUN
    NVIDIA_PYTORCH_VERSION=21.06 PYTORCH_BUILD_VERSION=1.9.0a0+c3d40fd 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 pytorch-quantization
    06/17/2021 11:02 PM 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
    06/17/2021 11:01 PM UTC
    sha256:6aee476e8278cb7dce2e6863c14c26afae068db139b89a31796b4d9f533169eaRUN
    NVIDIA_PYTORCH_VERSION=21.06 PYTORCH_BUILD_VERSION=1.9.0a0+c3d40fd 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
    06/17/2021 11:01 PM UTC
    sha256:4cc02d73abbdd49452a78d7c023f805d5b9f4da970c2401394ef394523419591COPY
    NVIDIA_Deep_Learning_Container_License.pdf /workspace/
    06/17/2021 11:01 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
    /usr/local/bin/nvidia_entrypoint.sh
    06/17/2021 11:01 PM UTC
    ...