44 lines
No EOL
1.4 KiB
Bash
Executable file
44 lines
No EOL
1.4 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
# Build script for Docker image with NVDEC hardware acceleration support
|
|
|
|
echo "Building Docker image with NVDEC hardware acceleration support..."
|
|
echo "========================================================="
|
|
|
|
# Build the base image first (with all ML and hardware acceleration dependencies)
|
|
echo "Building base image with NVDEC support..."
|
|
docker build -f Dockerfile.base -t detector-worker-base:nvdec .
|
|
|
|
if [ $? -ne 0 ]; then
|
|
echo "Failed to build base image"
|
|
exit 1
|
|
fi
|
|
|
|
# Build the main application image
|
|
echo "Building application image..."
|
|
docker build -t detector-worker:nvdec .
|
|
|
|
if [ $? -ne 0 ]; then
|
|
echo "Failed to build application image"
|
|
exit 1
|
|
fi
|
|
|
|
echo ""
|
|
echo "========================================================="
|
|
echo "Build complete!"
|
|
echo ""
|
|
echo "To run the container with GPU support:"
|
|
echo "docker run --gpus all -p 8000:8000 detector-worker:nvdec"
|
|
echo ""
|
|
echo "Hardware acceleration features enabled:"
|
|
echo "- NVDEC for H.264/H.265 video decoding"
|
|
echo "- NVENC for video encoding (if needed)"
|
|
echo "- TurboJPEG for fast JPEG encoding"
|
|
echo "- CUDA for model inference"
|
|
echo ""
|
|
echo "The application will automatically detect and use:"
|
|
echo "1. GStreamer with NVDEC (NVIDIA GPUs)"
|
|
echo "2. FFMPEG with CUVID (NVIDIA GPUs)"
|
|
echo "3. VAAPI (Intel/AMD GPUs)"
|
|
echo "4. TurboJPEG (3-5x faster than standard JPEG)"
|
|
echo "=========================================================" |