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:97d7d811f1b046c1dfdd0a8bc47f263863b5ab18400e6db5361e46274a527589RUN
    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
    		10/19/2020 3:24 PM UTC
    sha256:c8725fd1629c3544f2cb78b6ea61ade8c2d062cf2d4e5b46f0caf911b19cbe67RUN
    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"
    		10/19/2020 3:19 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-10.2/targets/aarch64-linux/lib:
    		10/19/2020 3:19 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
    		10/19/2020 3:19 PM UTC
    sha256:3e9098193eb02dd03e9e8abe41d45762a78680fc9ee14fd28551eada6ebdda76RUN
    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
    		10/19/2020 3:19 PM UTC
    sha256:863e90653906415944758c001766166107ff36a89066c33b1ebfd5b4b285ebafRUN
    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/*
    		10/19/2020 3:15 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCHAUDIO_VERSION
    		10/19/2020 3:15 PM UTC
    sha256:adfb60e46ee1687592c7e50207a6b38fa595b9321c10262ddff062152297eba7RUN
    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}"
    		10/19/2020 3:15 PM UTC
    sha256:948ac0abde329a2cd98856e9f8193c0a20ff6a3bce3c4c03d9eec9e27ab2a5c3RUN
    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/*
    		10/19/2020 3:03 PM UTC
    sha256:e0d1fc183e52a90350a96e47a7317d32d0f18315f56b0f021b52f1d3327ec76cRUN
    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"
    		10/19/2020 3:03 PM UTC