File tree Expand file tree Collapse file tree 1 file changed +38
-0
lines changed
Expand file tree Collapse file tree 1 file changed +38
-0
lines changed Original file line number Diff line number Diff line change 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"]
You can’t perform that action at this time.
0 commit comments