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:5157f3aaa0491a3797937b21051e97da9e545f3a2f335aa43f06eeb0dae8d4c4RUN
    OPENCV_DEB=OpenCV-4.5.0-aarch64.tar.gz OPENCV_URL=https://nvidia.box.com/shared/static/2hssa5g3v28ozvo3tc3qwxmn78yerca9.gz PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install --no-cache-dir --verbose pycuda six
    		08/04/2022 6:08 PM UTC
    sha256:5d1a39e7fb75170231f4394299c2dde94ed835567756ddd76d235eab80075f5dRUN
    OPENCV_DEB=OpenCV-4.5.0-aarch64.tar.gz OPENCV_URL=https://nvidia.box.com/shared/static/2hssa5g3v28ozvo3tc3qwxmn78yerca9.gz PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 echo "$PATH" &&
  echo "$LD_LIBRARY_PATH"
    		08/04/2022 6:01 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64:
    		08/04/2022 6:01 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PATH=/usr/local/cuda/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
    		08/04/2022 6:01 PM UTC
    sha256:0e651afae557b766ce9801749b4198778a495ca74463a884c2d343b4c50ce659RUN
    OPENCV_DEB=OpenCV-4.5.0-aarch64.tar.gz OPENCV_URL=https://nvidia.box.com/shared/static/2hssa5g3v28ozvo3tc3qwxmn78yerca9.gz PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 cd /tmp &&
  ./opencv_install.sh ${OPENCV_URL} ${OPENCV_DEB}
    		08/04/2022 6:01 PM UTC
    sha256:fe175acea3d15f2b2f536d48a504dd9e0eed366dd672214274df72e6ac330e4cCOPY
    file:738b23d3effc8e9ffc54a87257f6eadd655f97205a53b370e43cf91ac36a2748 in /tmp/opencv_install.sh
    		08/04/2022 6:00 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    OPENCV_DEB=OpenCV-4.5.0-aarch64.tar.gz
    		08/04/2022 6:00 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    OPENCV_URL=https://nvidia.box.com/shared/static/5v89u6g5rb62fpz4lh0rz531ajo2t5ef.gz
    		08/04/2022 6:00 PM UTC
    sha256:7fea22c659c1632b4b41541b3bbf1910dde703a7abc3ee42abcb671e99c94faaRUN
    PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.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 &&
  sed -i 's#  URL https://zlib.net/zlib-1.2.11.tar.gz#  URL https://zlib.net/zlib-1.2.12.tar.gz#g' third_party/zlib/CMakeLists.txt ||
  echo "failed to patch torchaudio/third_party/zlib/CMakeLists.txt" &&
  sed -i 's#  URL_HASH SHA256=c3e5e9fdd5004dcb542feda5ee4f0ff0744628baf8ed2dd5d66f8ca1197cb1a1#  URL_HASH SHA256=91844808532e5ce316b3c010929493c0244f3d37593afd6de04f71821d5136d9#g' third_party/zlib/CMakeLists.txt ||
  echo "failed to patch torchaudio/third_party/zlib/CMakeLists.txt" &&
  BUILD_SOX=1 python3 setup.py install &&
  cd ../ &&
  rm -rf torchaudio
    		08/04/2022 6:00 PM UTC
    sha256:834bf6dfd9568f0ccdb7ef65e3fac86e1b771a2eb2ca96635b72b2a691748b61RUN
    PYTORCH_URL=https://nvidia.box.com/shared/static/ssf2v7pf5i245fk4i0q926hy4imzs2ph.whl PYTORCH_WHL=torch-1.11.0-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install --no-cache-dir scikit-build &&
  pip3 install --no-cache-dir ninja &&
  pip3 install --no-cache-dir --verbose pysoundfile
    		08/04/2022 5:40 PM UTC