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:71cbf190c2069617e2b993689da1042e47ed3de1998d9ac19e2270e63d1e4197RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl PYTORCH_WHL=torch-1.8.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.8.0 TORCHVISION_VERSION=v0.9.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install --no-cache-dir --verbose pycuda six
    		07/27/2021 8:52 PM UTC
    sha256:f4845e67aa4627d9ec812089094308e7424cbedb0b529546720c98c0c2b0b539RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl PYTORCH_WHL=torch-1.8.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.8.0 TORCHVISION_VERSION=v0.9.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 echo "$PATH" &&
  echo "$LD_LIBRARY_PATH"
    		07/27/2021 8:47 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-10.2/targets/aarch64-linux/lib:
    		07/27/2021 8:47 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
    		07/27/2021 8:47 PM UTC
    sha256:e6873bfd659323be4e69e14eed9a0c2d97e0a05519256e07e1e583cd83342aa9RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl PYTORCH_WHL=torch-1.8.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.8.0 TORCHVISION_VERSION=v0.9.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 git clone --recursive -b ${TORCHAUDIO_VERSION} https://github.com/pytorch/audio torchaudio &&
  cd torchaudio &&
  python3 setup.py install &&
  cd ../ &&
  rm -rf torchaudio
    		07/27/2021 8:47 PM UTC
    sha256:ddf0b0cee9770dfe979ad625dc1101ab36a24ae847652e83d5c8120233e7eb4fRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl PYTORCH_WHL=torch-1.8.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.8.0 TORCHVISION_VERSION=v0.9.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install --no-cache-dir scikit-build &&
  pip3 install --no-cache-dir ninja
    		07/27/2021 8:43 PM UTC
    sha256:d67dd886a34f2f35e5631480c6b08837377f92376c246840634c7b85ac593345RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl PYTORCH_WHL=torch-1.8.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.8.0 TORCHVISION_VERSION=v0.9.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 apt-get update &&
  apt-get install -y --no-install-recommends cmake sox libsox-dev libsox-fmt-all &&
  rm -rf /var/lib/apt/lists/* &&
  apt-get clean
    		07/27/2021 8:41 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCHAUDIO_VERSION
    		07/27/2021 8:41 PM UTC
    sha256:1a27e94a953e65339657493fb32d460cee51724a8f74e234f1ff2233189ff6c6RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl PYTORCH_WHL=torch-1.8.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.9.0 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 --no-cache-dir "${PILLOW_VERSION}"
    		07/27/2021 8:41 PM UTC
    sha256:210ec66a4ed745243ed4eb65f98005c7ccd89ae0a0722bc83f77868318a9679eRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl PYTORCH_WHL=torch-1.8.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.9.0 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/* &&
  apt-get clean
    		07/27/2021 8:17 PM UTC