search

Kaixhin / dockerfiles

Compilation of Dockerfiles with automated builds enabled on the Docker Registry
https://hub.docker.com/u/kaixhin/
MIT License
505 stars 127 forks source link

Add documentation for cuda-caffe #40

Closed dreamflasher closed 6 years ago

dreamflasher commented 6 years ago

When running nvidia-docker run -it kaixhin/cuda-caffe:8.0 one ends up in a ~/caffe folder. But caffee seems not to be available via caffe command. Also it looks like that caffe is actually not built. How is one supposed to use this image?

Kaixhin commented 6 years ago

The Caffe documentation is here. The caffe binary is at ./build/install/bin/caffe (assuming the current directory is ~/caffe). $CAFFE_ROOT isn't set because you're placed in ~/caffe by default. I've just downloaded the image and gone through the LeNet/MNIST tutorial on GPU, so can confirm that it Caffe is installed correctly.