cuda dockerfile

deci committed May 20, 2025 at 12:50 UTC 44d133f8c66b333706acf16f35a712e755edc82e
1 file changed +20
docker/run/docker-compose.cuda.yml new
+20
@@ -0,0 +1,20 @@
1 +services:
2 + agent-zero-cuda:
3 + container_name: agent-zero-cuda
4 + image: frdel/agent-zero-run-cuda:testing
5 + volumes:
6 + - ./agent-zero:/a0
7 + - ./agent-zero/work_dir:/root
8 + ports:
9 + - "50080:80"
10 + environment:
11 + - NVIDIA_VISIBLE_DEVICES=all
12 + - NVIDIA_DRIVER_CAPABILITIES=compute,utility
13 + - PYTHONUNBUFFERED=1
14 + deploy:
15 + resources:
16 + reservations:
17 + devices:
18 + - driver: nvidia
19 + count: all
20 + capabilities: [gpu]
\ No newline at end of file