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:c295965641fc23daf9c8daeb4cead037840ad57873ac343ec97efed409bfcf0eRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl PYTORCH_WHL=torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0-rc1 TORCHVISION_VERSION=v0.7.0-rc2 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 pip3 install pycuda --verbose
    		07/02/2020 5:34 PM UTC
    sha256:24714fd420e291ef142468201024fca800db7ded19fa98f0515c276bb6fd1c9aRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl PYTORCH_WHL=torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0-rc1 TORCHVISION_VERSION=v0.7.0-rc2 TORCH_CUDA_ARCH_LIST=5.3;6.2;7.2 echo "$PATH" &&
  echo "$LD_LIBRARY_PATH"
    		07/02/2020 5:28 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-10.2/targets/aarch64-linux/lib:
    		07/02/2020 5:28 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
    		07/02/2020 5:28 PM UTC
    sha256:495fe771aa23273b7b1559502632d6c615133bf4e64bc7bde22576c70e58e178RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl PYTORCH_WHL=torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0-rc1 TORCHVISION_VERSION=v0.7.0-rc2 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
    		07/02/2020 5:28 PM UTC
    sha256:4c225d22910265fa1326e9a3cbedddd2f922678a835326e31a166453748b9129RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl PYTORCH_WHL=torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl TORCHAUDIO_VERSION=v0.6.0-rc1 TORCHVISION_VERSION=v0.7.0-rc2 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/*
    		07/02/2020 5:25 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCHAUDIO_VERSION
    		07/02/2020 5:25 PM UTC
    sha256:99bfdd15706b6b6e1f8447b5387988335270987b5460831a99e059fc2ef169bbRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl PYTORCH_WHL=torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.7.0-rc2 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}"
    		07/02/2020 5:25 PM UTC
    sha256:997fd4303142953f6ba743b00f1701e5e284a0c5e1c3c7646b6f6767cf3537aeRUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl PYTORCH_WHL=torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.7.0-rc2 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/*
    		07/02/2020 5:12 PM UTC
    sha256:ad5d71b110fa1a1ef9758a7d66d19b903c5efa73f3d5784ab622f7b1f8bfb1d2RUN
    PILLOW_VERSION=pillow PYTORCH_URL=https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl PYTORCH_WHL=torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl TORCHVISION_VERSION=v0.7.0-rc2 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"
    		07/02/2020 5:12 PM UTC