You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
439 B
Docker
24 lines
439 B
Docker
FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
|
|
#FROM tensorflow/tensorflow:2.12.0-gpu
|
|
|
|
ARG USER
|
|
ARG GROUP
|
|
ARG UID
|
|
ARG GID
|
|
|
|
RUN groupadd --gid ${GID} ${GROUP}
|
|
RUN useradd --shell /bin/bash --uid ${UID} --gid ${GID} --create-home ${USER}
|
|
RUN mkdir /wd
|
|
RUN chown ${USER}:${GROUP} /wd
|
|
|
|
# SYSTEM INITIALIZATION
|
|
# RUN pip install jupyter
|
|
|
|
USER ${USER}
|
|
|
|
# USER INITIALIZATION
|
|
# RUN ...
|
|
|
|
SHELL ["/bin/bash", "--login", "-i", "-c"]
|
|
WORKDIR /wd
|