This repository provides a recipe - in the from of instruction and Dockerfile for running GPU caffe workloads out of a docker container.
- Configure nvidia docker: follow these instructions.
- Verify your nvidia docker configuration:
sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
- Build the docker image from the dockerfile included in this repo:
docker build -t caffe .
- Run it:
run -it --runtime=nvidia --shm-size=8g --rm caffe
- Verify all is well by running through one of the caffe examples.