Issues
This commit is contained in:
+1
-2
@@ -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
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user