SAM-2 Serving as Docker Rest Endpoint

Before starting

Build the model with GPU requires nvidia base image. Follow the instructions before in order to login into nvcr.io

1. Be sure you have Docker installed and buildkit available
2. Connect to [NVIDIA NGC](https://org.ngc.nvidia.com/setup/personal-keys)
3. Create a personal key
4. Login into nvcr.io

Build and Run

Version

export version=x.y.z

GPU

bash install.sh  sam-2 gpu registry

docker run --gpus all -p 8080:8080 -e PORT=8080 -v ./huggingface:/root/.cache/huggingface/ sam-2-gpu:${version}

CPU

bash install.sh  sam-2 cpu registry

docker run -p 8080:8080 -e PORT=8080 -v ./huggingface:/root/.cache/huggingface/ sam-2-cpu:${version}