The instruction of installing Pytorch and TorchVision by Python3.8 on Ubuntu 20.04 on Jetson device

Hi all,

Here are my successful steps of installing Pytorch and TorchVision by Python3.8 on Ubuntu 20.04 (in container) on AGX / NX devices.

For the Pytorch part, I referred to this comment and it can work well in my environment. However, it didn’t mention how to install the Torchvision part. Hence, you can try to use these commands to install torchvision.

The steps of building pytorch v1.7

git clone --recursive --branch 1.7
cd pytorch
python3.8 -m pip install -r requirements.txt
python3.8 install

The building time will take almost over 6 hours on arm devices.

Here are my partial outputs.

byte-compiling /usr/local/lib/python3.8/dist-packages/caffe2/proto/ to hsm_pb2.cpython-38.pyc
byte-compiling /usr/local/lib/python3.8/dist-packages/caffe2/proto/ to torch_pb2.cpython-38.pyc
byte-compiling /usr/local/lib/python3.8/dist-packages/caffe2/proto/ to caffe2_legacy_pb2.cpython-38.pyc
byte-compiling /usr/local/lib/python3.8/dist-packages/caffe2/proto/ to metanet_pb2.cpython-38.pyc
byte-compiling /usr/local/lib/python3.8/dist-packages/caffe2/proto/ to caffe2_pb2.cpython-38.pyc
running install_egg_info
running egg_info
writing torch.egg-info/PKG-INFO
writing dependency_links to torch.egg-info/dependency_links.txt
writing entry points to torch.egg-info/entry_points.txt
writing requirements to torch.egg-info/requires.txt
writing top-level names to torch.egg-info/top_level.txt
reading manifest file 'torch.egg-info/SOURCES.txt'
writing manifest file 'torch.egg-info/SOURCES.txt'
removing '/usr/local/lib/python3.8/dist-packages/torch-1.7.0a0.egg-info' (and everything under it)
Copying torch.egg-info to /usr/local/lib/python3.8/dist-packages/torch-1.7.0a0.egg-info
running install_scripts
Installing convert-caffe2-to-onnx script to /usr/local/bin
Installing convert-onnx-to-caffe2 script to /usr/local/bin

And test in other places.

The steps of building torchvision (should be with v0.8)

The cuda version is 10.2.

export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64
export PATH=$PATH:$CUDA_HOME/bin

sudo apt-get install libjpeg-dev zlib1g-dev
git clone
git checkout v0.8.1-rc1
sudo -H python3 install


Test it:

Note: That one was installed in the container. My host of the jetson device was still OS 18.04 and python 3.6. (JetPack v4.4)

Thanks for the sharing.

1 Like