File size: 453 Bytes
b585c7f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
version: '3'
services:
h2ogpt:
build:
context: .
dockerfile: Dockerfile
restart: always
shm_size: '2gb'
ports:
- '${H2OGPT_PORT}:7860'
volumes:
- cache:/workspace/.cache
- save:/workspace/save
command: ${H2OGPT_ARGS}
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
volumes:
cache:
save:
|