Skip to content

Commit 1bff288

Browse files
Create Dockerfile.gpu3
1 parent f983912 commit 1bff288

File tree

1 file changed

+38
-0
lines changed

1 file changed

+38
-0
lines changed

automated_build/Dockerfile.gpu3

Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
FROM floydhub/dl-docker:gpu_temp2
2+
3+
MAINTAINER Sai Soundararaj <saip@outlook.com>
4+
5+
ARG THEANO_VERSION=rel-0.8.2
6+
ARG TENSORFLOW_VERSION=0.8.0
7+
ARG TENSORFLOW_ARCH=gpu
8+
ARG KERAS_VERSION=1.0.3
9+
ARG LASAGNE_VERSION=v0.1
10+
ARG TORCH_VERSION=latest
11+
ARG CAFFE_VERSION=master
12+
13+
14+
# Install the latest versions of nn, cutorch, cunn, cuDNN bindings and iTorch
15+
RUN luarocks install nn && \
16+
luarocks install cutorch && \
17+
luarocks install cunn && \
18+
\
19+
cd /root && git clone https://github.com/soumith/cudnn.torch.git && cd cudnn.torch && \
20+
git checkout R4 && \
21+
luarocks make && \
22+
\
23+
cd /root && git clone https://github.com/facebook/iTorch.git && \
24+
cd iTorch && \
25+
luarocks make
26+
27+
28+
# Set up notebook config
29+
COPY jupyter_notebook_config.py /root/.jupyter/
30+
31+
# Jupyter has issues with being run directly: https://github.com/ipython/ipython/issues/7062
32+
COPY run_jupyter.sh /root/
33+
34+
# Expose Ports for TensorBoard (6006), Ipython (8888)
35+
EXPOSE 6006 8888
36+
37+
WORKDIR "/root"
38+
CMD ["/bin/bash"]

0 commit comments

Comments
 (0)