ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
  • Meta
    NVIDIA L4T PyTorch
    Container
    Meta
    NVIDIA L4T PyTorch

    PyTorch is a GPU accelerated tensor computational framework with a Python front end. This container contains PyTorch and torchvision pre-installed in a Python 3 environment to get up & running quickly with PyTorch on Jetson.

  • LayerLabelCreated
    sha256:66bcdc238401212f9a73f9b12fe95ba9f385cca8bd87d31af64c048ebd272b9eRUN
    /bin/bash
    		01/21/2021 8:49 PM UTC
    sha256:f30dd4ea24f09244dc6a4f6559856802c37b1503d1c09f7bc2ff122e56a37b79RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl PYTORCH_WHL=torch-1.7.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.7.0 TORCHVISION_VERSION=v0.8.1 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install pycuda --verbose
    		12/22/2020 8:46 PM UTC
    sha256:4b7e04ab7aceb6631fbc61c8eda3469375ed7dd70457f1e0664d1923c7daeb56RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl PYTORCH_WHL=torch-1.7.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.7.0 TORCHVISION_VERSION=v0.8.1 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 echo "$PATH" &&
  echo "$LD_LIBRARY_PATH"
    		12/22/2020 8:42 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-10.2/targets/aarch64-linux/lib:
    		12/22/2020 8:42 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PATH=/usr/local/cuda/bin:/usr/local/cuda-10.2/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
    		12/22/2020 8:42 PM UTC
    sha256:c5458d5c46d63dcffff62f53d9cdddd52da5d4c1c67344aa4d4d3d8bd328ebb1RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl PYTORCH_WHL=torch-1.7.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.7.0 TORCHVISION_VERSION=v0.8.1 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 git clone -b ${TORCHAUDIO_VERSION} https://github.com/pytorch/audio torchaudio &&
  cd torchaudio &&
  python3 setup.py install &&
  cd ../ &&
  rm -rf torchaudio
    		12/22/2020 8:42 PM UTC
    sha256:f4efaaf0a4484cc629c9f1a48e674204733be2e456c8b389f2f651ecb4726267RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl PYTORCH_WHL=torch-1.7.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.7.0 TORCHVISION_VERSION=v0.8.1 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 apt-get update &&
  apt-get install -y --no-install-recommends sox libsox-dev libsox-fmt-all &&
  rm -rf /var/lib/apt/lists/*
    		12/22/2020 8:37 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCHAUDIO_VERSION
    		12/22/2020 8:37 PM UTC
    sha256:9fc338f2fa9b092f3a98543ae6cba519b5d5d9e521615f20b80b4984203e5287RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl PYTORCH_WHL=torch-1.7.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.8.1 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 git clone -b ${TORCHVISION_VERSION} https://github.com/pytorch/vision torchvision &&
  cd torchvision &&
  python3 setup.py install &&
  cd ../ &&
  rm -rf torchvision &&
  pip3 install "${PILLOW_VERSION}"
    		12/22/2020 8:37 PM UTC
    sha256:4ff84ca23dc9bc7e88281f5b7dae33b3b13c5dd9f5aa3809a231888b8f1a61fdRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl PYTORCH_WHL=torch-1.7.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.8.1 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 apt-get update &&
  apt-get install -y --no-install-recommends git build-essential libjpeg-dev zlib1g-dev &&
  rm -rf /var/lib/apt/lists/*
    		12/22/2020 8:19 PM UTC