56 lines
1.8 KiB
Plaintext
56 lines
1.8 KiB
Plaintext
# Use an existing image as the base
|
|
# FROM nvcr.io/nvidia/pytorch:23.04-py3
|
|
# FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-devel
|
|
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
|
|
#
|
|
# Add the current user to the image
|
|
ARG USER_NAME=lhunor
|
|
ARG USER_ID=11017
|
|
ARG GROUP_NAME=guest
|
|
ARG GROUP_ID=11000
|
|
|
|
# fix for tzdata asking for user input
|
|
ARG DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC
|
|
|
|
# Update the system and install required packages
|
|
RUN apt-get update && apt-get install -y sudo git apt-utils htop ffmpeg libsm6 libxext6
|
|
|
|
# Create a new group with the same GID as the host
|
|
RUN groupadd --gid $GROUP_ID $GROUP_NAME
|
|
|
|
# Create a new user with the same UID and GID as the host
|
|
RUN useradd --uid $USER_ID --gid $GROUP_ID --shell /bin/bash --create-home $USER_NAME
|
|
|
|
# don't require password with sudo, for convenience
|
|
# not the safest thing to do, but hopefully okay inside the container
|
|
RUN echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers
|
|
|
|
# Add the new user to the sudo group
|
|
RUN usermod -aG sudo $USER_NAME
|
|
|
|
# install conda stuff still as root
|
|
RUN conda update -n base conda
|
|
RUN conda install -c fvcore -c iopath -c conda-forge fvcore iopath -y
|
|
RUN conda install -c bottler nvidiacub -y
|
|
RUN conda install pytorch3d -c pytorch3d -y
|
|
|
|
# Set the default user to the new user
|
|
USER $USER_NAME
|
|
|
|
WORKDIR /workspace
|
|
|
|
ENV PATH="$PATH:/home/$USER_NAME/.local/bin"
|
|
RUN pip install --upgrade pip
|
|
COPY requirements.txt /tmp/
|
|
RUN pip install torch-cluster -f https://data.pyg.org/whl/torch-2.0.0+cu117.html
|
|
RUN pip install --requirement /tmp/requirements.txt
|
|
|
|
RUN pip install pyvista flask==2.1.2 pyinstaller==5.1 sphinx ipyvtklink pythreejs tqdm imageio imageio-ffmpeg matplotlib cmocean
|
|
|
|
# defining the volumes that will be mounted
|
|
VOLUME [ "/data-net" ]
|
|
VOLUME [ "/data-local" ]
|
|
VOLUME [ "/home-net" ]
|
|
|
|
# Start the container with a bash shell
|
|
CMD ["/bin/bash"] |