pytorch/docker
Paul Jesse Hellemn 8c6c09ad41 Adding openmpi to all conda builds (#2089)
* Adding openmpi to all conda builds

* Typo and turning off quiet

* Removing openmpi from non_cuda conda build

* Actually openmpi is already in the images
2018-03-01 09:19:34 -08:00
..
jenkins Adding openmpi to all conda builds (#2089) 2018-03-01 09:19:34 -08:00
ubuntu-14.04-cpu-all-options Docker move 2017-08-09 13:54:17 -07:00
ubuntu-14.04-cpu-minimal git clone from the master branch in the docker files because branch v0.8.1 does not exist 2018-01-30 21:21:01 -08:00
ubuntu-16.04-cpu-all-options Remove duplicate Docker dependency 2017-11-02 14:01:27 -07:00
ubuntu-16.04-cpu-minimal git clone from the master branch in the docker files because branch v0.8.1 does not exist 2018-01-30 21:21:01 -08:00
ubuntu-16.04-cuda8-cudnn6-all-options git clone from the master branch in the docker files because branch v0.8.1 does not exist 2018-01-30 21:21:01 -08:00
ubuntu-16.04-cuda8-cudnn7-all-options git clone from the master branch in the docker files because branch v0.8.1 does not exist 2018-01-30 21:21:01 -08:00
ubuntu-16.04-gpu-tutorial Docker move 2017-08-09 13:54:17 -07:00
readme.md Docker move 2017-08-09 13:54:17 -07:00

Docker & Caffe2

Note: use nvidia-docker to run all GPU builds.

To get the latest source, rerun the docker builds using the Dockerfiles.

Docker images at https://hub.docker.com/r/caffe2ai/caffe2/ are a few months old, but will be refreshed soon.

Build like: docker build -t caffe2:cuda8-cudnn6-all-options .

Run like: nvidia-docker run --rm -it caffe2:cuda8-cudnn6-all-options python -m caffe2.python.operator_test.relu_op_test

For Docker on USB related instructions you can find some help on the gh-pages branch here