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=33af6b134f1ac712b7c697cc66f218e97c360ded
    07/05/2022 10:20 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF
    07/05/2022 10:20 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=40241807
    07/05/2022 10:20 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=40241807
    07/05/2022 10:20 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID
    07/05/2022 10:20 AM UTC
    sha256:f1b23b80d9c490308c0aca9d6f39f5326a2c9c9f42083b674de8666de3a560adCOPY
    entrypoint.d/ /opt/nvidia/entrypoint.d/
    07/05/2022 10:20 AM UTC
    sha256:12ab04221f039058913cc4a4c615d54708859d2ff29cbd908c4fecb009e06183RUN
    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
    07/05/2022 10:20 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
    07/05/2022 10:20 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
    07/05/2022 10:20 AM UTC
    sha256:240cdd2f19e4653d9eb510fc0b4d990c0fbd02898a0c8a0d2f8f9b3ea51faa69RUN
    PYVER=3.8 pip install --no-cache-dir /opt/pytorch/torch_tensorrt/py/dist/*.whl
    07/05/2022 10:20 AM UTC
    ...