50d0f2fa1e
* Added NVIDIA GPU support to Docker * feat: Added NVIDIA GPU support to Docker --------- Co-authored-by: Abdellah Derfoufi <mohammed-abdellah.derfoufi@capgemini.com>
20 lines
482 B
YAML
20 lines
482 B
YAML
version: "3.8"
|
|
services:
|
|
rvc:
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
container_name: rvc
|
|
volumes:
|
|
- ./weights:/app/assets/weights
|
|
- ./opt:/app/opt
|
|
# - ./dataset:/app/dataset # you can use this folder in order to provide your dataset for model training
|
|
ports:
|
|
- 7865:7865
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu] |