Torch 2.0.1 for Nvidia Jetson AGX Orin Dev Kit 64GB

dusty_nv · August 2, 2023, 8:25pm

@iqbal.rahmoon I believe these message imply that you are trying to install torchvision/torchaudio via pip and their wheels on PyPi, which won’t work anyways because those PyPi wheels are built without CUDA support. Instead, see the Installation section of this post to build torchvision from source, or see the l4t-pytorch dockerfile to see how I build them:

github.com

dusty-nv/jetson-containers/blob/74afe69b91ed8256e1cd344a6c52301a36e631bc/Dockerfile.pytorch#L61


      
          ARG PYTORCH_URL
          ARG PYTORCH_WHL
          
          RUN wget --quiet --show-progress --progress=bar:force:noscroll --no-check-certificate ${PYTORCH_URL} -O ${PYTORCH_WHL} && \
              pip3 install --no-cache-dir --verbose ${PYTORCH_WHL} && \
              rm ${PYTORCH_WHL}
          
          RUN python3 -c 'import torch; print(f"PyTorch version: {torch.__version__}"); print(f"CUDA available:  {torch.cuda.is_available()}"); print(f"cuDNN version:   {torch.backends.cudnn.version()}"); print(torch.__config__.show());'
          
          
          #
          # torchvision
          #
          ARG TORCHVISION_VERSION
          ARG TORCH_CUDA_ARCH_LIST="5.3;6.2;7.2;8.7"
          
          RUN printenv && echo "torchvision version = $TORCHVISION_VERSION" && echo "TORCH_CUDA_ARCH_LIST = $TORCH_CUDA_ARCH_LIST"
          
          RUN apt-get update && \
              apt-get install -y --no-install-recommends \
          		  git \

Topic		Replies	Views
Torchvision on Jetson AGX Orin DevKit Jetson AGX Orin pytorch	7	2980	June 2, 2023
What is the compatible torchvision version for torch-1.14.0a0+44dac51c.nv23.2 installed on AGX ORIN( Ubuntu-22.04, JP-5.1/5.0, Python-3.8, CUDA-11.4) Jetson AGX Orin pytorch	19	3310	May 11, 2023
Is there a Pip installable version of TorchVision for Jetson AGX Orin? Jetson AGX Orin pytorch	8	2034	January 18, 2023
Torchvision version Jetpack 6.0 Jetson Orin Nano pytorch	2	444	August 27, 2024
PyTorch for Jetson Orin Jetpack 5.1 installation issue Jetson AGX Orin pytorch	12	1152	January 23, 2024
Jetpack6.1 install torchvision Jetson AGX Orin cuda , pytorch	20	696	October 30, 2024
Cannot get Torch met CUDA to work on Jetson Orin Jetson Orin NX pytorch	10	1768	February 26, 2024
Jetson Orin Nx, Pytorch 2.1.2 on Jetpack 6.0 DP, torch.cuda.is_available() is false Jetson Orin NX pytorch	8	2075	January 30, 2024
Help with Pytorch, torchvision on Jetpack 6 Jetson Orin Nano pytorch	7	3161	March 5, 2024
Torch and torchvision versions are not compatible Jetson AGX Orin pytorch	9	5249	September 11, 2023

Torch 2.0.1 for Nvidia Jetson AGX Orin Dev Kit 64GB

Related topics