-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
107 lines (86 loc) · 2.66 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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
FROM nvidia/cuda:8.0-cudnn5-devel
MAINTAINER Jan Deriu <deri@zahw.ch>
ARG THEANO_VERSION=0.9.0
ARG TENSORFLOW_VERSION=1.1.0
ARG TENSORFLOW_ARCH=gpu
ARG KERAS_VERSION=2.0.4
ARG LASAGNE_VERSION=v0.1
ARG TORCH_VERSION=latest
# Install some dependencies
RUN apt-get update && apt-get install -y \
bc \
build-essential \
cmake \
curl \
g++ \
gfortran \
git \
libffi-dev \
libfreetype6-dev \
libhdf5-dev \
libjpeg-dev \
liblcms2-dev \
libopenblas-dev \
liblapack-dev \
libpng12-dev \
libssl-dev \
libtiff5-dev \
libwebp-dev \
libzmq3-dev \
nano \
pkg-config \
software-properties-common \
unzip \
vim \
wget \
zlib1g-dev \
&& \
apt-get clean && \
apt-get autoremove && \
rm -rf /var/lib/apt/lists/* && \
# Link BLAS library to use OpenBLAS using the alternatives mechanism (https://www.scipy.org/scipylib/building/linux.html#debian-ubuntu)
update-alternatives --set libblas.so.3 /usr/lib/openblas-base/libblas.so.3
RUN apt-get upgrade
RUN wget http://repo.continuum.io/archive/Anaconda3-4.4.0-Linux-x86_64.sh
RUN ls
RUN sh -c '/bin/echo -e "\nyes\n" |bash Anaconda3-4.4.0-Linux-x86_64.sh'
ENV PATH "$PATH:/root/anaconda3/bin"
RUN conda install numpy scipy mkl nose sphinx
RUN conda install theano pygpu
#Install Theano
RUN echo "[global]\ndevice=cuda0\nfloatX=float32\noptimizer_including=cudnn\nmode=FAST_RUN \
\n[lib]\ncnmem=0.95 \
\n[nvcc]\nfastmath=True" \
> /root/.theanorc
# Install TensorFlow
RUN pip install --ignore-installed --upgrade tensorflow-gpu==1.2
# Install Keras
RUN pip install --no-cache-dir --upgrade keras
RUN mkdir /root/.keras/
RUN touch /root/.keras/keras.json
RUN pip install --no-cache-dir --upgrade nltk
RUN pip install --no-cache-dir --upgrade tqdm
RUN pip install --no-cache-dir --upgrade gensim
RUN python -m nltk.downloader punkt
RUN python -m nltk.downloader stopwords
RUN python -m nltk.downloader nonbreaking_prefixes
RUN python -m nltk.downloader perluniprops
RUN mkdir /DLFramework
COPY vae_architectures /DLFramework/vae_architectures
COPY keras_fit_utils /DLFramework/keras_fit_utils
COPY custom_layers /DLFramework/custom_layers
COPY vae_gan_architectures /DLFramework/vae_gan_architectures
COPY sc_lstm_architecutre /DLFramework/sc_lstm_architecutre
COPY data_loaders /DLFramework/data_loaders
COPY preprocessing_utils.py /DLFramework
COPY output_text.py /DLFramework
COPY custom_callbacks.py /DLFramework
COPY main.py /DLFramework
COPY main_hybrid.py /DLFramework
COPY main_hybrid_gan.py /DLFramework
COPY main_cornell.py /DLFramework
COPY main_gan_cornell.py /DLFramework
COPY main_nlg_sclstm.py /DLFramework
COPY main_nlg_scvae.py /DLFramework
WORKDIR /DLFramework
CMD python3 main.py -c config_vae.json