From 09fff16d2a11e2e2ca1ad46ef47cf1ea5592006e Mon Sep 17 00:00:00 2001 From: Mher Kazandjian Date: Fri, 27 Nov 2020 01:36:44 +0100 Subject: [PATCH] install char-rnn and add the cmd to launch termianl in that workdir --- CUDA/10.0-ubuntu18.04/Dockerfile | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/CUDA/10.0-ubuntu18.04/Dockerfile b/CUDA/10.0-ubuntu18.04/Dockerfile index a6e31ba..de02c5b 100644 --- a/CUDA/10.0-ubuntu18.04/Dockerfile +++ b/CUDA/10.0-ubuntu18.04/Dockerfile @@ -13,7 +13,8 @@ RUN apt-get -y install \ python-pip \ git \ libhdf5-dev \ - software-properties-common + software-properties-common \ + libssl-dev # Fix 'sudo: command not found' # https://github.com/torch/distro/blob/master/install-deps contains 'sudo', but nvidia:cuda removed sudo: https://github.com/crisbal/docker-torch-rnn/issues/9#issuecomment-365362656 @@ -86,5 +87,11 @@ RUN cd /root/torch/extra/cutorch/ && luarocks make rocks/cutorch-scm-1.rockspec #RUN luarocks install cutorch RUN luarocks install cunn +RUN cd /root && git clone https://github.com/karpathy/char-rnn.git +#RUN cd /root/char-rnn && th train.lua -data_dir data/some_folder -rnn_size 512 -num_layers 2 -dropout 0.5 + #Done! -WORKDIR /root/torch-rnn +WORKDIR /root/char-rnn + +# docker run --runtime=nvidia mherkazandjian/char-rnn:latest th train.lua -data_dir data/tinyshakespeare -rnn_size 512 -num_layers 2 -dropout 0.5 +CMD ["/bin/bash --rcfile ~/.bashrc"]