pytorch/docker/caffe2
Johannes M Dieterich 4ee9c56218 Support PyTorch ROCm CI on Ubuntu18.04 (#31886)
Summary:
In order to support Ubuntu18.04, some changes to the scripts are required.
* install dependencies with -y flag
* mark install noninteractive
* install some required dependencies (gpg-agent, python3-distutils, libidn11)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/31886

Differential Revision: D19300586

Pulled By: bddppq

fbshipit-source-id: d7fb815a3845697ce63af191a5bc449d661ff1de
2020-01-07 10:32:47 -08:00
..
jenkins Support PyTorch ROCm CI on Ubuntu18.04 (#31886) 2020-01-07 10:32:47 -08:00
ubuntu-14.04-cpu-all-options Move Caffe2 Dockerfiles to docker/caffe2 (#2430) 2018-03-27 15:51:50 -07:00
ubuntu-14.04-cpu-minimal Move Caffe2 Dockerfiles to docker/caffe2 (#2430) 2018-03-27 15:51:50 -07:00
ubuntu-16.04-cpu-all-options Move Caffe2 Dockerfiles to docker/caffe2 (#2430) 2018-03-27 15:51:50 -07:00
ubuntu-16.04-cpu-minimal Move Caffe2 Dockerfiles to docker/caffe2 (#2430) 2018-03-27 15:51:50 -07:00
ubuntu-16.04-cuda8-cudnn6-all-options Move Caffe2 Dockerfiles to docker/caffe2 (#2430) 2018-03-27 15:51:50 -07:00
ubuntu-16.04-cuda8-cudnn7-all-options fix caffe2 docker build (#7411) 2018-06-05 22:51:43 -04:00
ubuntu-16.04-gpu-tutorial Remove tutorials from main repository. 2018-03-29 09:31:09 -07:00
readme.md Move Caffe2 Dockerfiles to docker/caffe2 (#2430) 2018-03-27 15:51:50 -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