run-35
This commit is contained in:
+7
-28
@@ -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
|
||||
Reference in New Issue
Block a user