mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
e908cf28f4
Summary: Bringing over selected dockerfiles from documentation branch and updated the GPU Dockerfiles to use some of lukeyeager provided docker configurations. Latest docker with CUDA 8.0 and cuDNN 6 can be pulled via `docker pull caffe2ai/caffe2` or built with `ubuntu-16.04-cuda8-cudnn6-all-options/Dockerfile`. **You must use nvidia-docker instead of docker to run the GPU-enabled dockers.** Tutorial files can be overlaid by building `ubuntu-16.04-gpu-tutorial/Dockerfile`. Supersedes #911. Closes #876. Closes #923. Closes https://github.com/caffe2/caffe2/pull/949 Reviewed By: Yangqing Differential Revision: D5510872 Pulled By: aaronmarkham fbshipit-source-id: 390f5eea1d9ec1a3edda828470b12386ab8a1775 |
||
|---|---|---|
| .. | ||
| ubuntu-14.04-cpu-all-options | ||
| ubuntu-14.04-cpu-minimal | ||
| ubuntu-16.04-cpu-all-options | ||
| ubuntu-16.04-cpu-minimal | ||
| ubuntu-16.04-cuda8-cudnn6-all-options | ||
| ubuntu-16.04-cuda8-cudnn7-all-options | ||
| ubuntu-16.04-gpu-tutorial | ||
| readme.md | ||
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