Nvidia-docker-compose
Simple wrapper for docker-compose to use GPU enabled docker under nvidia-docker
Installing
To install the script, simply run:
Example
version: '3.3'
services:
core:
image: your/proj-core:1.1.1-ubuntu1804-gpu
devices:
- /dev/nvidia0
- /dev/nvidiactl
- /dev/nvidia-uvm
restart: always
network_mode: host
volumes:
- /home/your/core/node:/usr/local/c2core/storage/node
- /home/your/core/python:/usr/local/c2core/storage/python
- /home/your/core/workspace:/usr/local/c2core/storage/workspace
command:
- --verbose
- --sync
- node
entrypoint:
- /usr/local/c2core/bin/c2node