-
Notifications
You must be signed in to change notification settings - Fork 21
/
Dockerfile
20 lines (16 loc) · 777 Bytes
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
FROM pytorch/pytorch:1.3-cuda10.1-cudnn7-runtime
WORKDIR /workspace
ENV PYTHONPATH /workspace
ADD requirements.txt /workspace/
RUN pip install -r requirements.txt
ADD configs /workspace/configs/
ADD homographies /workspace/homographies/
ADD inference /workspace/inference/
ADD runfiles /workspace/runfiles/
ADD superres /workspace/superres/
ADD *.py /workspace/
ADD results/checkpoints/256/ /workspace/results/checkpoints/256/
ADD results/encoders/256_mlp/models/ /workspace/results/encoders/256_mlp/models/
ADD results/encoders/256_resnet/models/ /workspace/results/encoders/256_resnet/models/
ADD results/pretrained_models/ /workspace/results/pretrained_models/
ADD https://download.pytorch.org/models/vgg19-dcbb9e9d.pth /root/.cache/torch/checkpoints/vgg19-dcbb9e9d.pth