-
Notifications
You must be signed in to change notification settings - Fork 62
/
Dockerfile
38 lines (26 loc) · 1.34 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
FROM pytorch/pytorch:1.1.0-cuda10.0-cudnn7.5-devel
COPY . /workspace/VLP
# set python3 to default
RUN echo "alias python=python3" >> /root/.bashrc && \
bash /root/.bashrc
# install dependencies
RUN apt update -y
RUN apt install wget vim zip unzip ca-certificates-java openjdk-8-jdk -y
WORKDIR /workspace/VLP
RUN bash ./setup.sh
# install python libraries
RUN pip install tensorboardX six numpy tqdm pandas scikit-learn py-rouge matplotlib scikit-image h5py demjson torchtext stanfordnlp # --user
RUN pip install -e git://github.com/Maluuba/nlg-eval.git#egg=nlg-eval # --user
RUN cd ./coco-caption && ./get_stanford_models.sh
# detectron fc7 weights
RUN wget https://dl.fbaipublicfiles.com/ActivityNet-Entities/ActivityNet-Entities/detectron_weights.tar.gz && tar -zxvf detectron_weights.tar.gz && rm detectron_weights.tar.gz
RUN cd ./pythia && \
mkdir -p data && cd data && \
wget http://dl.fbaipublicfiles.com/pythia/data/vocab.tar.gz && \
tar xf vocab.tar.gz && rm vocab.tar.gz
RUN cd ./pythia/data && wget https://s3.amazonaws.com/cvmlp/vqa/mscoco/vqa/v2_Annotations_Val_mscoco.zip && \
unzip v2_Annotations_Val_mscoco.zip && rm v2_Annotations_Val_mscoco.zip
RUN cd ./pythia/data && mkdir -p imdb && cd imdb && \
wget https://dl.fbaipublicfiles.com/pythia/data/imdb/vqa.tar.gz && \
tar xf vqa.tar.gz && rm vqa.tar.gz
EXPOSE 8888