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=2235e97cfdf0c73e377b69c5420a10e4efa84fe7
    02/19/2025 10:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF=2235e97cfdf0c73e377b69c5420a10e4efa84fe7
    02/19/2025 10:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=143088498
    02/19/2025 10:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=143088498
    02/19/2025 10:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID=143088498
    02/19/2025 10:56 PM UTC
    sha256:2e871af64be31e1a5da9898435ce1f787392042a512d414bd822f90a43a96efdRUN
    NVIDIA_PYTORCH_VERSION=25.02 PYTORCH_BUILD_VERSION=2.7.0a0+ecf3bae NVFUSER_BUILD_VERSION=6627725 TARGETARCH=arm64 PYVER=3.12 PYVER_MAJMIN=312 L4T=1 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali cd /opt/pytorch/pytorch/third_party/gloo &&
      patch -p1 < /gloo.patch &&
      cd - &&
      rm /gloo.patch
    02/19/2025 10:56 PM UTC
    sha256:220ec27642a903d386701402dadaaf5068db6380a1c4b0f32413625af6837c7fCOPY
    gloo.patch /gloo.patch
    02/19/2025 10:56 PM UTC
    sha256:2a67ac56754c79832dec3bd69d7cd59200b4f90f27c2216c25f31096689a66c3COPY
    entrypoint.d/ /opt/nvidia/entrypoint.d/
    02/19/2025 10:56 PM UTC
    sha256:131c6a636d46c53c9093599e43d4479bb125fea641d3d66038c82bde2e7ee90dRUN
    NVIDIA_PYTORCH_VERSION=25.02 PYTORCH_BUILD_VERSION=2.7.0a0+ecf3bae NVFUSER_BUILD_VERSION=6627725 TARGETARCH=arm64 PYVER=3.12 PYVER_MAJMIN=312 L4T=1 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali 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
    02/19/2025 10:56 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    TORCH_NCCL_USE_COMM_NONBLOCKING=0
    02/19/2025 10:56 PM UTC
    ...