From 89d93a5a20a8a6a39c8d7282192cb4e5dde6562c Mon Sep 17 00:00:00 2001
From: Ryan <ryan@rymon.net>
Date: Mon, 10 Aug 2026 20:24:40 -0400
Subject: [PATCH] Issues

---
 docker-compose.yml | 3 +--
 worker/Dockerfile  | 3 ++-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 74af8d0..1304ada 100644
--- a/docker-compose.yml
+++ b/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:
diff --git a/worker/Dockerfile b/worker/Dockerfile
index e4558a8..3ba6fa5 100644
--- a/worker/Dockerfile
+++ b/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