File size: 1,864 Bytes
82b1bc7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
# Build: sudo docker build -t <project_name> .
# Run: sudo docker run -v $(pwd):/workspace/project --gpus all -it --rm <project_name>
ARG USERNAME=kitt
ARG USER_UID=1005
ARG USER_GID=100
FROM nvidia/cuda:12.1.1-base-ubuntu22.04
ENV PYTHON_VERSION=3.10
ENV PATH /opt/conda/bin:$PATH
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/lib"
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/conda/lib"
ENV PYTHONIOENCODING=UTF-8
ENV LANG=C.UTF-8
ENV LC_ALL=C.UTF-8
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
ENV DEBIAN_FRONTEND=noninteractive
ENV CONDA_AUTO_UPDATE_CONDA=false
RUN apt update
RUN apt install -y bash \
build-essential \
git \
curl \
ca-certificates \
wget \
&& rm -rf /var/lib/apt/lists
RUN echo $USERNAME
RUN echo $USER_UID
RUN echo $USER_GID
# Create the user
RUN groupadd -f --gid 100 users \
&& useradd --uid 1005 --gid 100 -m kitt -s /bin/bash
RUN mkdir /opt/conda \
&& chown -R kitt:users /opt/conda
USER kitt
# Install Miniconda and create main env
ADD --chown=kitt:users --chmod=744 https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh /tmp/miniconda3.sh
RUN ls -laxoh .
RUN /bin/bash /tmp/miniconda3.sh -b -u -p /opt/conda \
&& rm /tmp/miniconda3.sh \
&& /opt/conda/bin/conda install -y -c anaconda \
python=$PYTHON_VERSION \
&& /opt/conda/bin/conda clean -ya
RUN /opt/conda/bin/conda config --set ssl_verify False \
&& pip install --upgrade pip --trusted-host pypi.org --trusted-host files.pythonhosted.org \
&& echo ". /opt/conda/etc/profile.d/conda.sh" >> ~/.bashrc \
&& echo "conda activate base" >> ~/.bashrc
# && ln -s /opt/conda/bin/pip /usr/local/bin/pip3
# Install requirements
COPY --chown=kitt:users requirements.txt /tmp
RUN pip install --no-cache-dir -r /tmp/requirements.txt \
&& rm /tmp/requirements.txt
CMD ["/bin/bash"]
|