Issues

docker-compose.yml

@@ -31,7 +31,6 @@ services:
       dockerfile: worker/Dockerfile
     container_name: videodetect-worker
     restart: unless-stopped
-    runtime: nvidia
     environment:
       - NVIDIA_VISIBLE_DEVICES=all
       - NVIDIA_DRIVER_CAPABILITIES=compute,utility
@@ -55,7 +54,7 @@ services:
           devices:
             - driver: nvidia
               count: all
-              capabilities: [gpu]
+              capabilities: ["gpu"]
         limits:
           memory: 24G
     networks:

worker/Dockerfile

@@ -1,5 +1,6 @@
 # Base image: CUDA 11.8 runtime on Ubuntu 22.04
-FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
+#FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
+FROM nvidia/cuda:13.3.1-cudnn-runtime-ubuntu22.04
 
 # Avoid interactive prompts during build
 ENV DEBIAN_FRONTEND=noninteractive