Instalación de Pytorch en NVIDIA Jetson - jamarma/tfm-muva GitHub Wiki

Información

En esta guía se explica cómo instalar Pytorch en las tarjetas NVIDIA Jetson. La tarjeta usada es una Jetson Xavier NX con JetPack 5.1.2.

Instalación de Torch

Dependencias necesarias.

sudo apt-get -y update
sudo apt-get install -y python3-pip libopenblas-dev

pip install numpy==1.23.1

Elegir versión de torch.

export TORCH_INSTALL=https://developer.download.nvidia.com/compute/redist/jp/v$JP_VERSION/pytorch/$PYT_VERSION
  • JP_VERSION: versión de JetPack. Para JetPack 5.1.1 usar v511
  • PYT_VERSION: versión de Pytorch. Mirar la tabla de compatibilidad.

Para JetPack 5.1 y torch 2.0.0 ejecutar lo siguiente.

export TORCH_INSTALL=https://developer.download.nvidia.cn/compute/redist/jp/v511/pytorch/torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl

Instalar torch.

pip install --no-cache $TORCH_INSTALL

Instalación de torchvision

Dependencias necesarias.

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev

Instalar torchvision.

pip install torchvision==0.15.1