Skip to content

Instantly share code, notes, and snippets.

@Ayke
Created February 14, 2020 23:57
Show Gist options
  • Save Ayke/4eeb5d3dd8798e419ae7b20e30ac51f6 to your computer and use it in GitHub Desktop.
Save Ayke/4eeb5d3dd8798e419ae7b20e30ac51f6 to your computer and use it in GitHub Desktop.
Configure Detectron 2 on nvidia TX2

Install Detectron 2

As it says here, https://github.com/facebookresearch/detectron2/blob/master/INSTALL.md we need to install a few modules before installing detectron.

The only part can be difficult is to install pytorch on TX2 since TX2 is aarch64 architecture, the installation instructions on pytorch website do not work. We need to install via this page: https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/

To upgrade pyyaml, just pip3 install pyyaml --ignore-installed

Also you might want to keep using python3 and pip3 to avoid using python2

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