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:c0839d8d16becfff4a9aa23607733d458b9be045244a7c6c55e3fe90bdc5ec81RUN
    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
    		04/01/2022 6:27 PM UTC
    sha256:027faa2c2aa1919cb607c5eb3558ee3afede1890f408ba9ca895d3ae713c41b3RUN
    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"
    		04/01/2022 6:24 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64:
    		04/01/2022 6:24 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
    		04/01/2022 6:24 PM UTC
    sha256:f323b3050cdd59092f2c4af09b2417af1aa6172545328f4591e8f01ca3d70f1aRUN
    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
    		04/01/2022 6:24 PM UTC
    sha256:2050e56b9b1df7335a608c431b31c44547b5395bada05b1b60d9facdfe99b7bcRUN
    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
    		04/01/2022 6:20 PM UTC
    sha256:a791c0161136780bdabf3ffb3c09291167ec352c1f354ebdd928ba3bb593fad7RUN
    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
    		04/01/2022 6:19 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCHAUDIO_VERSION
    		04/01/2022 6:19 PM UTC
    sha256:d051fe0a8833419e0f18c768439a5d3b5e13608f0f84ab77e2b6ed5016281f3cRUN
    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}
    		04/01/2022 6:19 PM UTC
    sha256:a17e2fcfb740fce6dd913565e0f19139e1a034bcddcdbf440fc7472ae084d0d0RUN
    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
    		04/01/2022 6:19 PM UTC