Skip to content

FEAT: Add Docker support#4

Open
andreas-loeffler wants to merge 7 commits intofacebookresearch:mainfrom
andreas-loeffler:feat/docker_support
Open

FEAT: Add Docker support#4
andreas-loeffler wants to merge 7 commits intofacebookresearch:mainfrom
andreas-loeffler:feat/docker_support

Conversation

@andreas-loeffler
Copy link
Copy Markdown

FEAT: Add Docker support

This PR is adding Docker support for running Boxer on a Host Machine based on a Nvidia GPU

Changes

Docker integration:

  • docker/Dockerfile to define a docker env with all dependencies, GPU support
  • docker/docker-compose.yml: container orchestration with GPU passthrough + mounting of ckpts, sample_data and output dirs
  • docker/entrypoint.sh: auto-downloads model checkpoints and sample Aria data on first run (skipped on subsequent runs)
  • scripts/boxer_docker.sh: docker container helper for running commands inside the Docker container, including X11 forwarding
  • .dockerignore: to exclude checkpoints, sample data, and outputs from the build context

3D viewer output dir

  • utils/viewer_3d.py: updated to use the EVAL_PATH env variable for the output dir

How to Test

Note: The setup was tested/developed on a Host Machine running Ubuntu 24.04 with a NVIDIA GPU with driver version 580.126.09 and the NVIDIA Container Toolkit version 1.19.0

Build the image

docker build -f docker/Dockerfile -t boxer .

Headless Test

bash scripts/boxer_docker.sh python run_boxer.py --input nym10_gen1 --max_n=90 --track --skip_viz

GUI test

bash scripts/boxer_docker.sh python view_prompt.py --input nym10_gen1

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Meta Open Source bot. label Apr 11, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant