Docker container for running CDEEP3m - CUDA 9.0 - OpenCV 2 - CuDNN 7
Dependencies:
To run execute the following steps:
- Collect and prepare training data
mkdir train
cp -r 'TRAININGFILES AND FOLDERS' train/
- Build docker compose
docker-compose build
- Edit and replace COMMAND in docker-compose.yml
version: '2.3'
services:
cdeep3m:
build:
dockerfile: Dockerfile
context: .
image: cdeep3m
runtime: nvidia
environment:
- NVIDIA_VISIBLE_DEVICES=all
command: COMMAND # runtraining.sh COMMAND
volumes:
- ./train:/train
- Run:
docker-compose up
- If rebuilding docker-compose image after changes to cdeep3m master run:
docker-compose build --no-cache