-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
33 lines (24 loc) · 1.14 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
FROM kaixhin/cuda-caffe:8.0
MAINTAINER Socratic Datum <socratic.datum@gmail.com>
# Move to root and clone PPGN repo
RUN cd /root/ && git clone https://github.com/Evolving-AI-Lab/ppgn.git
# Clone the caffe_lrcn repo for image captioning (example 5)
RUN cd /root/ && git clone https://github.com/anguyen8/caffe_lrcn.git
RUN cd /root/ && git clone https://github.com/thommiano/caffe_lrcn_joel.git
# Move into models folder and download
RUN cd /root/ppgn/nets/generator/noiseless && ./download.sh
RUN cd /root/ppgn/nets/caffenet && ./download.sh
RUN cd /root/ppgn/nets/placesCNN && ./download.sh
RUN cd /root/ppgn/nets/lrcn && ./download.sh
# Update packages and install image magic for command-line interfacees
RUN apt-get update && apt-get -y install imagemagick
# Install nano
RUN apt-get update && apt-get -y install nano
# Install virtualenv
RUN pip install virtualenv
# Create a Python 3 environment in the ppgn directory
RUN cd /root/ppgn && virtualenv -p python3 env
# Edit file path for caffe/python
RUN cd /root/ppgn && sed -i "s|/path/to/your/caffe/python|/root/caffe/python |g" settings.py
# Launch in ppgn directory/root/ppgn
WORKDIR /root/ppgn