Added NVIDIA GPU support to Docker (#1231)

* Added NVIDIA GPU support to Docker

* feat: Added NVIDIA GPU support to Docker

---------

Co-authored-by: Abdellah Derfoufi <mohammed-abdellah.derfoufi@capgemini.com>
This commit is contained in:
Abdellah derfoufi
2023-09-12 17:18:04 +02:00
committed by GitHub
parent e4e2f7f1ed
commit 50d0f2fa1e
2 changed files with 29 additions and 4 deletions

View File

@@ -10,4 +10,11 @@ services:
- ./opt:/app/opt
# - ./dataset:/app/dataset # you can use this folder in order to provide your dataset for model training
ports:
- 7865:7865
- 7865:7865
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]