Skip to content

Conversation

windreamer
Copy link
Collaborator

  • build: enable torch 2.8.0 and torchvision 0.23.0
  • build(docker): add jetson docker image

Motivation:

This PR adds a Docker image specifically built for Jetson devices, enabling easier development and deployment on these platforms. Currently, we are simulating the Jetson environment on arm64 servers using the NVIDIA L4T (Linux for Tegra) base image.

Modification:

A new Dockerfile is added to build an image compatible with the Jetson architecture. This image leverages the PyTorch wheel packages built with https://github.com/dusty-nv/jetson-containers to ensure compatibility and optimized performance on Jetson devices.

Checklist:

  • Pre-commit/linting tools used.
  • Modification covered by test workflow.
  • Documentation updated.

@windreamer windreamer force-pushed the try_jetson_docker branch 2 times, most recently from 24e0d82 to 9f3e432 Compare August 14, 2025 02:13
@windreamer windreamer mentioned this pull request Sep 9, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant