Skip to content

Instantly share code, notes, and snippets.

@winnerineast
Forked from Ayke/DetectrononTX2.md
Created May 20, 2020 02:23
Show Gist options
  • Select an option

  • Save winnerineast/6b6256fb87ad78f2834a204150c147a2 to your computer and use it in GitHub Desktop.

Select an option

Save winnerineast/6b6256fb87ad78f2834a204150c147a2 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