df84c69676
* Docker image for test and experiment Keras - Docker image with CUDA support on ubuntu 14.04 - nvidia-docker script to forward the GPU to the container - MakeFile to simplify docker commands for build, run, test, ..etc - Add useful tools like jupyter notebook, ipdb, sklearn for experiments * update nvidia-docker plugin * use .theanorc in Dockerfile * Add tensorflow to the docker image * update Docker image to cuDNN v5 * test fixes * move docker to sub directory * README for docker * Fix typos * Add visualization to Dockerfile
27 lines
697 B
Makefile
27 lines
697 B
Makefile
help:
|
|
@cat Makefile
|
|
|
|
DATA?="${HOME}/Data"
|
|
GPU?=0
|
|
DOCKER_FILE=Dockerfile
|
|
DOCKER=GPU=$(GPU) nvidia-docker
|
|
BACKEND=tensorflow
|
|
TEST=tests/
|
|
SRC=$(shell dirname `pwd`)
|
|
|
|
build:
|
|
docker build -t keras --build-arg python_version=3.5 -f $(DOCKER_FILE) .
|
|
|
|
bash: build
|
|
$(DOCKER) run -it -v $(SRC):/src -v $(DATA):/data --env KERAS_BACKEND=$(BACKEND) keras bash
|
|
|
|
ipython: build
|
|
$(DOCKER) run -it -v $(SRC):/src -v $(DATA):/data --env KERAS_BACKEND=$(BACKEND) keras ipython
|
|
|
|
notebook: build
|
|
$(DOCKER) run -it -v $(SRC):/src -v $(DATA):/data --net=host --env KERAS_BACKEND=$(BACKEND) keras
|
|
|
|
test: build
|
|
$(DOCKER) run -it -v $(SRC):/src -v $(DATA):/data --env KERAS_BACKEND=$(BACKEND) keras py.test $(TEST)
|
|
|