Skip to content

Commit

Permalink
fix(ci): Proper GPU forwarding in docker compose (#3473)
Browse files Browse the repository at this point in the history
## What ❔

Fix for GPU forwarding in docker compose

## Why ❔

Docker update broke old way of forwarding GPU

## Checklist

<!-- Check your PR fulfills the following items. -->
<!-- For draft PRs check the boxes as you complete them. -->

- [x] PR title corresponds to the body of PR (we generate changelog
entries from PRs).
- [x] Tests for the changes have been added / updated.
- [x] Documentation comments have been added / updated.
- [x] Code has been formatted via `zkstack dev fmt` and `zkstack dev
lint`.
  • Loading branch information
artmakh authored Jan 14, 2025
1 parent 770cad8 commit 8518b3e
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
4 changes: 3 additions & 1 deletion docker-compose-gpu-runner-cuda-12-0.yml
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,9 @@ services:
resources:
reservations:
devices:
- capabilities: [ gpu ]
- driver: nvidia
count: all
capabilities: [gpu]

postgres:
image: "postgres:14"
Expand Down
4 changes: 3 additions & 1 deletion docker-compose-gpu-runner.yml
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,9 @@ services:
resources:
reservations:
devices:
- capabilities: [ gpu ]
- driver: nvidia
count: all
capabilities: [gpu]
postgres:
image: "postgres:14"
command: postgres -c 'max_connections=200'
Expand Down

0 comments on commit 8518b3e

Please sign in to comment.