mirror of
https://github.com/rishikanthc/Scriberr.git
synced 2026-07-01 08:15:46 +00:00
9aafe13450
- Update docker-compose.gpu.yml to use Dockerfile-cuda128 instead of Dockerfile-gpu - Change Docker image reference from cuda-11 to cuda128 to align with the new CUDA version
19 lines
834 B
YAML
19 lines
834 B
YAML
# This can be added when running the main docker-compose.yml file to add gpu support
|
|
# add this in your command line: docker-compose -f docker-compose.yml -f docker-compose.gpu.yml up
|
|
services:
|
|
app:
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile-cuda128
|
|
# You can find your architecture by running: nvidia-smi if on linux
|
|
# You can find your architecture by running: system_profiler SPDisplaysDataType if on mac
|
|
# You can find your architecture by running: wmic path win32_videocontroller get name if on windows
|
|
# You will need to change the image to match your architecture, E.G. "main-cuda-11"
|
|
image: ghcr.io/rishikanthc/scriberr:main-cuda128
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: all
|
|
capabilities: [gpu] |