Skip to content

Conversation

@dgdgksj
Copy link

@dgdgksj dgdgksj commented Apr 9, 2025

Summary

This PR adds Docker support to the Samurai project:

  • Dockerfile: Based on pytorch/pytorch:2.5.1-cuda12.1-cudnn9-devel
  • docker-compose.yaml: Simplifies GPU-enabled container execution
  • Verified to work on Ubuntu 22.04 with NVIDIA GPU (native Linux environment)

Thanks!

Screenshots

📸 Screenshots

🏗️ 1. Docker Build Started

Initial build process using docker-compose build --no-cache
Shows base image pull and package installation.
1

✅ 2. Docker Build Completed

All checkpoints downloaded, build successfully finished and tagged.
2

🚀 3. Container Running & Script Executed

Running docker-compose run samurai and executing demo.py inside container.
Shows successful inference in SAMURAI mode.
3

📺 Demo Video

Here is a short demo showing the Docker container in action, running demo.py with test video:

▶️ Watch on YouTube

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