This container assumes python2 (and CUDA if needed) installed, and simply install all repo's dependencies and launch bash session.
-
Install docker.
-
Make sure
docker
daemon is running (sudo service docker start
). -
Build the project image
docker build . -t bm
- Run the container (
--rm
option will automatically remove container when it exits)
nvidia-docker run --rm --name=bm-container -it bm bash
This container builds CUDA, cuDNN and python before installing dependencies.
-
Install additionally nvidia-docker.
-
Build the project image
docker build . -t bm-cuda -f Dockerfile-gpu
- Run the container
nvidia-docker run --rm --name=bm-cuda-container -it bm-cuda