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
Meta
NVIDIA L4T PyTorch
Container
Meta
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:b49a62fe851604f3246f7d1527350b97700a949615760bdcf2fdc46e03722ab7RUN
PYTORCH_URL=https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl PYTORCH_WHL=torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 pip3 install --no-cache-dir --verbose pycuda six
05/23/2022 2:39 PM UTC
sha256:8c39ee6646abed094f9316fdb0e5b06091b9fa61f22c392c978b0c4cab9059beRUN
PYTORCH_URL=https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl PYTORCH_WHL=torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 echo "$PATH" &&
  echo "$LD_LIBRARY_PATH"
05/23/2022 2:37 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64:
05/23/2022 2:37 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
05/23/2022 2:37 PM UTC
sha256:5cac95e165e90875a1c236553f85dc8fe404b8c403a1aa06e0146491b6291c49RUN
PYTORCH_URL=https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl PYTORCH_WHL=torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 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
05/23/2022 2:37 PM UTC
sha256:84f295e714c568fa0008aa528a61ff940c271cec0a672144dbc40f50dcb491beRUN
PYTORCH_URL=https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl PYTORCH_WHL=torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 pip3 install --no-cache-dir scikit-build &&
  pip3 install --no-cache-dir ninja &&
  pip3 install --no-cache-dir --verbose pysoundfile
05/23/2022 2:32 PM UTC
sha256:fd460be645e4b520439bb397293df377bc75825aa64ce4d0b710931cb67094b7RUN
PYTORCH_URL=https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl PYTORCH_WHL=torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl TORCHAUDIO_VERSION=v0.11.0 TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 apt-get update &&
  apt-get install -y --no-install-recommends pkg-config libffi-dev libsndfile1 &&
  rm -rf /var/lib/apt/lists/* &&
  apt-get clean
05/23/2022 2:32 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
TORCHAUDIO_VERSION
05/23/2022 2:32 PM UTC
sha256:687c315625fe23efb58522fd6b9c3d45076eba201ebb894a4e5a947c8e690b1cRUN
PYTORCH_URL=https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl PYTORCH_WHL=torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 PYTHON_ROOT=`pip3 show torch | grep Location: | cut -d' ' -f2` &&
  TORCH_CMAKE_CONFIG=$PYTHON_ROOT/torch/share/cmake/Torch/TorchConfig.cmake &&
  echo "patching _GLIBCXX_USE_CXX11_ABI in ${TORCH_CMAKE_CONFIG}" &&
  sed -i 's/  set(TORCH_CXX_FLAGS "-D_GLIBCXX_USE_CXX11_ABI=")/  set(TORCH_CXX_FLAGS "-D_GLIBCXX_USE_CXX11_ABI=0")/g' ${TORCH_CMAKE_CONFIG}
05/23/2022 2:32 PM UTC
sha256:5ed0dddbcf625c8e1b1c722e31adc8e0ccf309be49087a1c801559c5af0d00c2RUN
PYTORCH_URL=https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl PYTORCH_WHL=torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl TORCHVISION_VERSION=v0.12.0 TORCH_CUDA_ARCH_LIST=7.2;8.7 cmake --version
05/23/2022 2:32 PM UTC