run-35

docker/Dockerfile

@@ -1,31 +1,10 @@
-FROM tensorflow/tensorflow:2.3.1-gpu
-# FROM tensorflow/tensorflow:2.4.0rc0-gpu
-MAINTAINER moono.song "toilety@gmail.com"
-RUN apt-get update -y
-RUN apt-get install -y build-essential
-RUN pip install pillow
+FROM tensorflow/tensorflow:2.9.0-gpu
 
-# add sudoer
-ARG USERNAME=anonymous
-ARG USER_UID=1000
-ARG USER_GID=$USER_UID
+RUN apt-get update
+RUN apt-get upgrade -y
+RUN apt-get install git ffmpeg libsm6 libxext6 libopenexr-dev zip unzip htop nano -y
+RUN pip install numpy matplotlib opencv-python wandb OpenEXR tensorflow-graphics scipy plotly pandas tensorboardX opencv-python
 
-# Create the user
-RUN groupadd --gid $USER_GID $USERNAME \
-    && useradd --uid $USER_UID --gid $USER_GID -m $USERNAME \
-    #
-    # [Optional] Add sudo support. Omit if you don't need to install software after connecting.
-    && apt-get update \
-    && apt-get install -y sudo \
-    && echo $USERNAME ALL=\(root\) NOPASSWD:ALL > /etc/sudoers.d/$USERNAME \
-    && chmod 0440 /etc/sudoers.d/$USERNAME
 
-# ********************************************************
-# * Anything else you want to do like clean up goes here *
-# ********************************************************
-
-# [Optional] Set the default user. Omit if you want to keep the default as root.
-USER $USERNAME
-
-# add port for multiworker training
-EXPOSE 12345
+WORKDIR /home/workspace/data
+# RUN wandb login

docker/requierements.txt

@@ -0,0 +1,11 @@
+numpy
+matplotlib
+opencv-python
+wandb
+OpenEXR
+tensorflow-graphics
+scipy
+plotly
+pandas
+tensorboardX
+opencv-python