Dockerfile

FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04

RUN apt-get update && apt-get install -y \
    git cmake build-essential python3 python3-pip python3-dev \
    libboost-all-dev libopenblas-dev liblapack-dev

WORKDIR /dlib
RUN git clone https://github.com/davisking/dlib.git . \
  && mkdir build && cd build \
  && cmake .. -DDLIB_USE_CUDA=1 \
  && cmake --build . --config Release

# install via pip so dependencies like wheel are used
RUN pip install --upgrade pip wheel cmake setuptools && \
    cd /dlib && python3 setup.py install

# Choose a writable directory, e.g., /home/user/huggingface
RUN mkdir -p /home/user/huggingface
RUN chmod -R 777 /home/user/huggingface

# Then set environment variables
ENV HF_HOME=/home/user/huggingface \
    HF_HUB_CACHE=/home/user/huggingface/hub \
    TRANSFORMERS_CACHE=/home/user/huggingface/hub

WORKDIR /app
COPY requirements.txt .
RUN pip install -r requirements.txt
COPY . .
CMD ["python3", "app.py"]


# FROM yogi0421/dlib-base-image:python38-dlib19.24.0

# WORKDIR /app

# # Choose a writable directory, e.g., /home/user/huggingface
# RUN mkdir -p /home/user/huggingface
# RUN chmod -R 777 /home/user/huggingface

# # Then set environment variables
# ENV HF_HOME=/home/user/huggingface \
#     HF_HUB_CACHE=/home/user/huggingface/hub \
#     TRANSFORMERS_CACHE=/home/user/huggingface/hub

# COPY requirements.txt .
# RUN pip install --upgrade pip \
#     && pip install -r requirements.txt

# COPY . .
# CMD ["python", "app.py"]