forked from zepspaiva/docker-neuraltalk2
-
Notifications
You must be signed in to change notification settings - Fork 12
/
Dockerfile
40 lines (32 loc) · 1.04 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
# Version 1.0.4
FROM ubuntu:14.04
MAINTAINER Enrique Otero "[email protected]"
RUN rm /bin/sh && ln -s /bin/bash /bin/sh
RUN apt-get update
RUN sudo apt-get -y install \
git \
build-essential \
cmake \
wget \
curl \
libatlas-base-dev \
gfortran
RUN curl -sk https://raw.githubusercontent.com/torch/ezinstall/master/install-deps | bash
RUN git clone https://github.com/torch/distro.git ~/torch --recursive
RUN cd ~/torch; ./install.sh
RUN /bin/bash -c "source ~/.bashrc"
ENV PATH="$PATH:/root/torch/install/bin"
RUN luarocks install nn
RUN luarocks install nngraph
RUN luarocks install image
# Only for training
RUN apt-get -y install libprotobuf-dev protobuf-compiler
RUN luarocks install loadcaffe
RUN apt-get -y install libhdf5-dev hdf5-tools python-dev python-pip
RUN pip install cython numpy h5py scipy
RUN luarocks install hdf5
WORKDIR /home
RUN git clone https://github.com/karpathy/neuraltalk2
WORKDIR /home/neuraltalk2
RUN wget http://cs.stanford.edu/people/karpathy/neuraltalk2/checkpoint_v1_cpu.zip
RUN unzip checkpoint_v1_cpu.zip