This commit is contained in:
Ryan
2026-08-10 20:24:40 -04:00
parent 9ae1cd9918
commit 89d93a5a20
2 changed files with 3 additions and 3 deletions
+1 -2
View File
@@ -31,7 +31,6 @@ services:
dockerfile: worker/Dockerfile dockerfile: worker/Dockerfile
container_name: videodetect-worker container_name: videodetect-worker
restart: unless-stopped restart: unless-stopped
runtime: nvidia
environment: environment:
- NVIDIA_VISIBLE_DEVICES=all - NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=compute,utility - NVIDIA_DRIVER_CAPABILITIES=compute,utility
@@ -55,7 +54,7 @@ services:
devices: devices:
- driver: nvidia - driver: nvidia
count: all count: all
capabilities: [gpu] capabilities: ["gpu"]
limits: limits:
memory: 24G memory: 24G
networks: networks:
+2 -1
View File
@@ -1,5 +1,6 @@
# Base image: CUDA 11.8 runtime on Ubuntu 22.04 # 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 # Avoid interactive prompts during build
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive