Initial commit
This commit is contained in:
20
darknet-master/docker-compose.yml
Normal file
20
darknet-master/docker-compose.yml
Normal file
@@ -0,0 +1,20 @@
|
||||
version: '2'
|
||||
|
||||
services:
|
||||
yolo-gpu:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: Dockerfile.gpu
|
||||
image: yolo:gpu
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
devices:
|
||||
- driver: nvidia
|
||||
count: 1
|
||||
capabilities: [gpu]
|
||||
yolo-cpu:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: Dockerfile.cpu
|
||||
image: yolo:cpu
|
||||
Reference in New Issue
Block a user