-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
50 lines (40 loc) · 1.27 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
FROM nvidia/cuda:8.0-cudnn5-devel
ENV PATH="/opt/conda/bin:${PATH}"
RUN \
apt-get update && \
apt-get -y install wget
RUN apt-get -y install bzip2
RUN echo 'export PATH=/opt/conda/bin:$PATH' > /etc/profile.d/conda.sh && \
wget --quiet https://repo.continuum.io/archive/Anaconda2-5.0.0-Linux-x86_64.sh -O ~/anaconda.sh && \
/bin/bash ~/anaconda.sh -b -p /opt/conda && \
rm ~/anaconda.sh
RUN \
conda install -y bcolz && \
conda upgrade -y --all
RUN \
# install and configure theano
pip install theano && \
echo "[global] \n\
device = gpu \n\
floatX = float32 \n\
[cuda] \n\
root = /usr/local/cuda" > ~/.theanorc
RUN \
# install and configure keras
pip install keras==1.2.2 && \
mkdir ~/.keras && \
echo '{ \n\
"image_dim_ordering": "th", \n\
"epsilon": 1e-07, \n\
"floatx": "float32", \n\
"backend": "theano" \n\
}' > ~/.keras/keras.json
RUN \
# jupyter nbextensions
conda install -c conda-forge jupyter_contrib_nbextensions && \
conda install -c conda-forge jupyter_nbextensions_configurator && \
# Enable favorite jupyter notebook extensions (https://github.com/ipython-contrib/jupyter_contrib_nbextensions#3-enablingdisabling-extensions)
jupyter nbextension enable toc2/main && \
pip install kaggle-cli
RUN \
apt-get install unzip