uas-at-ucla / suas-23-24

Software/Vision operations for UAS@UCLA, 2023-2024. For a list of tasks, visit https://github.com/orgs/uas-at-ucla/projects/1
3 stars 0 forks source link

Setup for Jetson Orin Nano #54

Closed nathanchan631 closed 7 months ago

nathanchan631 commented 7 months ago

Overview:

Jetson flashed with Jetpack 6.0 DP using sdkmanager. Username: jetson Password: uas-2024

Jetson requires a different torch wheel than the PyPi one to use GPU, which is achieved with the dustynv/l4t-pytorch:r36.2.0 docker image. A modified installation of ultralytics is also required to use this installation of torch. Here we reference the official ultralytics Dockerfile for jetson devices. To sync this up with the normal branch of development (for everyone not developing on the jetson), the original Dockerfile has been updated to use the ultralytics/ultralytics:8.1.10 image.

The official Loki plugin for Docker doesn't work for ARM, so an alternative has been provided, which can be installed with make build-arm

Version info:

Usage Notes:

To use GPU in Docker Containers:

If running on the Jetson, change the dockerfile on line 5 of docker-compose.yml to Dockerfile-jetson

To ssh into the jetson, connect via usb-c or ethernet and run ssh jetson@192.168.55.1

For performance (seems to help a little bit):