Skip to content

Instantly share code, notes, and snippets.

@soumith

soumith/pytorch-agx-xavier.sh

Last active August 8, 2021 12:33
Show Gist options
  • Star 5 You must be signed in to star a gist
  • Fork 3 You must be signed in to fork a gist
  • Save soumith/45f61db3379db67414814ad9d5f2a427 to your computer and use it in GitHub Desktop.
Save soumith/45f61db3379db67414814ad9d5f2a427 to your computer and use it in GitHub Desktop.
Download ZIP
Install PyTorch on NVIDIA AGX Xavier using JetPack 4.1
Raw
pytorch-agx-xavier.sh
sudo apt-get install -y libopenblas-dev cmake ninja-build
sudo apt-get install -y python-pip
sudo pip install virtualenv
virtualenv pytorch-env
. pytorch-env/bin/activate
git clone https://github.com/pytorch/pytorch --recursive
pip install -r requirements.txt
python setup.py install
@JarvisEQ
Copy link

Line 8 should be
pip install -r ./requirements.txt

@KastanDay
Copy link

KastanDay commented Jul 14, 2021 •
edited

I recommend using the official distribution for Jetson from Nvidia (for Jetson Nano, Jetson TX1/TX2, and Jetson Xavier NX/AGX with JetPack 4.2 and newer).
https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-9-0-now-available/72048

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment