init

Dockerfile

@@ -0,0 +1,11 @@
+FROM tensorflow/tensorflow:1.14.0-gpu-py3
+RUN pip install jupyter
+
+WORKDIR /data
+COPY requirements.txt ./
+RUN pip install --no-cache-dir -r requirements.txt
+
+EXPOSE 7860
+
+COPY . .
+CMD ["streamlit", "run", "app.py", "--server.port", "7860"]

app.py

@@ -0,0 +1,12 @@
+import streamlit as st
+import tensorflow as tf
+
+def check_gpu():
+    return tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)
+
+st.title('GPU Availability Checker')
+
+if check_gpu():
+    st.success('GPU is available!')
+else:
+    st.warning('GPU is not available.')

requirements.txt

@@ -0,0 +1,5 @@
+requests
+matplotlib
+tqdm
+ipywidgets
+streamlit