diff --git a/base/Dockerfile b/base/Dockerfile new file mode 100644 index 0000000..1cc5e62 --- /dev/null +++ b/base/Dockerfile @@ -0,0 +1,13 @@ +FROM nvidia/cuda:9.0-cudnn7-runtime + +MAINTAINER Newnius + +USER root + +RUN apt update && \ + apt install -y git + +ADD bootstrap.sh /etc/bootstrap.sh + +ENTRYPOINT ["/etc/bootstrap.sh"] + diff --git a/base/bootstrap.sh b/base/bootstrap.sh new file mode 100755 index 0000000..54c44e3 --- /dev/null +++ b/base/bootstrap.sh @@ -0,0 +1,14 @@ +#!/bin/bash + +if ! [[ -z "${repo}" ]]; then + if [[ -z "${branch}" ]]; then + git clone $repo workspace + else + git clone -b $branch $repo /workspace + fi +fi + +cd /workspace + +$@ + diff --git a/python3/Dockerfile b/python3/Dockerfile new file mode 100644 index 0000000..7a711f1 --- /dev/null +++ b/python3/Dockerfile @@ -0,0 +1,9 @@ +FROM quickdeploy/base + +MAINTAINER Newnius + +USER root + +RUN apt update && \ + apt install -y python3 python3-pip + diff --git a/tensorflow/1.12/Dockerfile b/tensorflow/1.12/Dockerfile new file mode 100644 index 0000000..76ebc58 --- /dev/null +++ b/tensorflow/1.12/Dockerfile @@ -0,0 +1,7 @@ +FROM quickdeploy/python3 + +MAINTAINER Newnius + +USER root + +RUN pip3 install tensorflow-gpu==1.12 numpy pandas matplotlib sklearn scipy