adl.Dockerfile 210 Bytes
Newer Older
qianyj's avatar
qianyj committed
1
2
3
4
5
6
7
8
# Dockerfile for building AdaptDL-enabled CIFAR10 image
# Set docker build context to current folder

FROM pytorch/pytorch:1.4-cuda10.1-cudnn7-runtime

RUN pip install nni adaptdl tensorboard

COPY ./ /cifar10