add tf 2.1

tensorflow/2.1-gpu/Dockerfile

@@ -0,0 +1,15 @@
+FROM tensorflow/tensorflow:2.1.0-gpu-py3
+
+MAINTAINER Newnius <newnius.cn@gmail.com>
+
+USER root
+
+RUN apt update && \
+	apt install -y git && \
+	rm -rf /var/lib/apt/lists/*
+
+RUN pip3 install requests psutil py-cpuinfo pandas typing
+
+ADD bootstrap.sh /etc/bootstrap.sh
+
+ENTRYPOINT ["/etc/bootstrap.sh"]

tensorflow/2.1-gpu/bootstrap.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+
+if ! [[ -z "${repo}" ]]; then
+  if [[ -z "${branch}" ]]; then
+    git clone $repo /workspace
+  else
+    git clone -b $branch $repo /workspace
+  fi
+fi
+
+if [ -d /workspace ]; then
+  cd /workspace
+fi
+
+sleep infinity
+
+# use eval because commands likes `key=value command` would cause file not found error when using $@, but this eval will ruin current environment
+eval $@
+