Update settings

Author: lucasdavid

.dockerignore

@@ -128,6 +128,8 @@ dmypy.json
 # Pyre type checker
 .pyre/
 
-data/
-logs/
+data**
+logs**
 
+.idea/
+.vscode/

.gitignore

@@ -130,7 +130,10 @@ dmypy.json
 # Pyre type checker
 .pyre/
 
-data/
+data
 logs/
 secrets/
 datasets
+
+.idea/
+.vscode/

README.md

@@ -1,4 +1,4 @@
-# [![Tensorflow Logo](data/tf.png)](https://tensorflow.org) Algorithms in Tensorflow
+# [![Tensorflow Logo](assets/tf.png)](https://tensorflow.org) Algorithms in Tensorflow
 
 Experiments with tensorflow 2.0, gpu support, persistent logging and stable docker env.
 

actions/install.sh

@@ -1,4 +1,4 @@
 #!/bin/bash
 
-apt-get -qq update
-apt-get -qq install -y python-pydot python-pydot-ng graphviz
+apt-get update
+apt-get install -y python-pydot python-pydot-ng graphviz libgl1-mesa-glx

actions/run.sh

@@ -1,17 +1,17 @@
-#!/bin/bash
+#!/bin/bash -ex
 
 run()
 {
   CMD=${1:-up}
 
   if [[ "$CMD" == "up" ]]; then
-    docker-compose --env-file config/.env -f config/docker-compose.yml up
+    docker-compose -f config/docker-compose.yml up
   elif [[ "$CMD" == "tensorboard" ]]; then
-    docker-compose --env-file config/.env -f config/docker-compose.yml \
+    docker-compose -f config/docker-compose.yml \
                  exec experiments \
                  tensorboard --logdir "${2:-/tf/logs/}" --bind_all
   else
-    docker-compose --env-file config/.env -f config/docker-compose.yml ${@:1}
+    docker-compose -f config/docker-compose.yml ${@:1}
   fi
 }
 

data/tf.png renamed to assets/tf.png

@@ -12,4 +12,4 @@ RUN pip install -qr /tmp/requirements.txt
 ADD config/notebook.json /root/.jupyter/nbconfig/notebook.json
 
 ARG jupyter_theme
-RUN jt -t $jupyter_theme
+RUN jt -t ${jupyter_theme}

config/Dockerfile

@@ -1,6 +1,9 @@
-IMAGE_BASE=tensorflow/tensorflow:latest-gpu-jupyter
-JUPYTER_THEME=grade3 -T -f roboto -fs 10
+RUNTIME=nvidia
+
+IMAGE_BASE=tensorflow/tensorflow:latest-jupyter
 IMAGE=ldavid/experiments
 NVIDIA_VISIBLE_DEVICES=all
 NVIDIA_DRIVER_CAPABILITIES=all
 TF_FORCE_GPU_ALLOW_GROWTH=true
+
+JUPYTER_THEME=grade3 -T -f roboto -fs 10

config/docker-compose.yml

@@ -0,0 +1,22 @@
+version: '3'
+services:
+  experiments:
+    runtime: "${RUNTIME}"
+    build:
+      context: .
+      dockerfile: config/Dockerfile
+      args:
+        base_image: "${IMAGE_BASE}"
+        jupyter_theme: "${JUPYTER_THEME}"
+    image: "${IMAGE}"
+    mem_limit: 12g
+    ports:
+      - 8888:8888
+      - 6006:6006
+    volumes:
+      - ./notebooks:/tf
+      - ./logs:/tf/logs
+      - ./data:/data
+      - ./data/cached/keras:/root/.keras
+      - ./data/cached/tfds:/root/tensorflow_datasets
+      - ./secrets/kaggle:/root/.kaggle