Skip to content

Commit 2933f30

Browse files
committed
Enable GPU access to container
1 parent 6044949 commit 2933f30

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

docker-compose.yml.example

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,3 +24,10 @@ services:
2424
init: true
2525
ipc: host
2626
network_mode: host
27+
deploy:
28+
resources:
29+
reservations:
30+
devices:
31+
- driver: nvidia
32+
count: 1
33+
capabilities: [gpu]

0 commit comments

Comments
 (0)