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=c80c372d5366ad38038bbca10b6e31589e76e5ea
    04/17/2022 3:02 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF
    04/17/2022 3:02 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=36527063
    04/17/2022 3:02 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=36527063
    04/17/2022 3:02 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID
    04/17/2022 3:02 AM UTC
    sha256:4d03e2fdfcaecee4a428a407c7ea8c6406682c77ae2341a5baff1983f80d2fbaCOPY
    entrypoint.d/ /opt/nvidia/entrypoint.d/
    04/17/2022 3:02 AM UTC
    sha256:2adb20a548b2ed53827843796fc36c8c0eb05a06545b599b13171bedc1751bc3RUN
    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
    04/17/2022 3:02 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PATH=/opt/conda/lib/python3.8/site-packages/torch_tensorrt/bin:/opt/conda/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
    04/17/2022 3:02 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/opt/conda/lib/python3.8/site-packages/torch/lib:/opt/conda/lib/python3.8/site-packages/torch_tensorrt/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
    04/17/2022 3:02 AM UTC
    sha256:686cdcb3779f4e91d18e8b318708604e5d950660d61d24ae9afc894597759197RUN
    PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
    04/17/2022 3:02 AM UTC
    ...