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:b9e9d3887268d964dc030981816b451786d6c21b5ccd923d3d733afab493f616RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0 TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install pycuda --verbose
    		12/07/2020 9:39 PM UTC
    sha256:c8219a7d4bea8d989e4733ec9ff71a8fbe2b5e5398cc3a738113b0036128de93RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0 TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 echo "$PATH" &&
  echo "$LD_LIBRARY_PATH"
    		12/07/2020 9:35 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-10.2/targets/aarch64-linux/lib:
    		12/07/2020 9:35 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/07/2020 9:35 PM UTC
    sha256:93b7b1478d26a17d7ba5678af4d6fd39f309db25f4b1ae8a7156a589e93ff286RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0 TORCHVISION_VERSION=v0.7.0 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/07/2020 9:35 PM UTC
    sha256:1885dddc79e815e731b35aa42197ec8c2776e01b72ed68a135046a805e77956fRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0 TORCHVISION_VERSION=v0.7.0 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/07/2020 9:31 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCHAUDIO_VERSION
    		12/07/2020 9:31 PM UTC
    sha256:561e500c4b69ca6a99924382e8fc3b42d2512864bc0deebf9adfe5d5831f7462RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.7.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 "${PILLOW_VERSION}"
    		12/07/2020 9:31 PM UTC
    sha256:e429652ff20241e7a58695362b57fa365d0debae9a61c63026d86b4c0c7795d3RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.7.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/*
    		12/07/2020 9:19 PM UTC
    sha256:786c90dad7299cc52d0dea478d51565337e641dac1efd54141fa7c65eac4b8c2RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl PYTORCH_WHL=torch-1.6.0-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.7.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 printenv &&
  echo "torchvision version = $TORCHVISION_VERSION" &&
  echo "pillow version = $PILLOW_VERSION" &&
  echo "TORCH_CUDA_ARCH_LIST = $TORCH_CUDA_ARCH_LIST"
    		12/07/2020 9:19 PM UTC