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=f30e309ddf0414af88159ece258986e2c21b9f6d
    06/11/2020 2:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF
    06/11/2020 2:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=13419386
    06/11/2020 2:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=13419386
    06/11/2020 2:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID
    06/11/2020 2:45 AM UTC
    sha256:9b441e1a8fca6c449dbc33395d6611016d53335c9058056d26c23c02885a7622RUN
    NVIDIA_PYTORCH_VERSION=20.06 PYTORCH_BUILD_VERSION=1.6.0a0+9907a3e PYVER=3.6 cd /opt/mellanox/DEBS &&
      rm -f 5.0-0 &&
      ln -sf 4.6-1.0.1 5.0-0 &&
      ln -sf 4.6-1.0.1 5.0-2.1.8
    06/11/2020 2:45 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
    06/11/2020 2:45 AM UTC
    sha256:4f29d4fc8251a76439f16882269bf73032d6a7c584c13bf04adf6d7f392bfb43RUN
    NVIDIA_PYTORCH_VERSION=20.06 PYTORCH_BUILD_VERSION=1.6.0a0+9907a3e PYVER=3.6 BASE=0 DEVEL=0 SAMPLES=0 PYTHON=1 /nvidia/build-scripts/installTRT.sh &&
      mv /usr/src/tensorrt /opt &&
      ln -s /opt/tensorrt /usr/src/tensorrt
    06/11/2020 2:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
    /usr/local/bin/nvidia_entrypoint.sh
    06/11/2020 2:45 AM UTC
    sha256:1f7957821bbab52903ba6ce33a8a30ed6deddf689dca3f7cfdbe89e8f14f2d22COPY
    nvidia_entrypoint.sh /usr/local/bin
    06/11/2020 2:45 AM UTC
    ...