-
Notifications
You must be signed in to change notification settings - Fork 97
/
Dockerfile
33 lines (25 loc) · 961 Bytes
/
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
# How to use:
# docker build --tag genre:latest .
# docker run --rm -it genre:latest /bin/bash
# docker run --rm -it -v $(pwd)/tests:/GENRE/genre/tests genre:latest /bin/bash
# pytest genre/tests
FROM python:3.8
WORKDIR /GENRE/
RUN apt-get update && \
apt-get install axel -y
RUN mkdir data && \
axel -n 20 http://dl.fbaipublicfiles.com/GENRE/kilt_titles_trie_dict.pkl -o data
RUN mkdir models && \
axel -n 20 http://dl.fbaipublicfiles.com/GENRE/fairseq_wikipage_retrieval.tar.gz && \
tar -xvf fairseq_wikipage_retrieval.tar.gz --directory models && \
rm fairseq_wikipage_retrieval.tar.gz
# Install PyTorch
RUN pip install torch --no-cache-dir
# Install dependencies
RUN pip install pytest requests --no-cache-dir
# Install fairseq
RUN git clone -b fixing_prefix_allowed_tokens_fn --single-branch https://github.com/nicola-decao/fairseq.git
RUN pip install -e ./fairseq
# Install genre
COPY . genre
RUN pip install -e ./genre