google-research

1
FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04
2

3
# Install the basics
4
RUN apt-get -y update --fix-missing
5
RUN apt-get install -y \
6
	emacs \
7
	htop \
8
	python2.7 \
9
	python-pip \
10
	git
11
RUN pip install --upgrade pip
12

13
# Add the source-code
14
RUN mkdir /home/state_of_sparsity
15
WORKDIR /home/state_of_sparsity
16
COPY . .
17

18
# Install the dependencies
19
RUN pip install -r requirements.txt
20

21
# Set the python path
22
RUN export PYTHONPATH=$PYTHONPATH:`pwd`
23