Skip to content

Add Dockerfile and docker-compose setup for containerized usage#115

Open
dgdgksj wants to merge 1 commit into
yangchris11:masterfrom
dgdgksj:add-docker-support
Open

Add Dockerfile and docker-compose setup for containerized usage#115
dgdgksj wants to merge 1 commit into
yangchris11:masterfrom
dgdgksj:add-docker-support

Conversation

@dgdgksj

@dgdgksj dgdgksj commented Apr 9, 2025

Copy link
Copy Markdown

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