|
|
|
|
|
|
|
|
FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04 |
|
|
|
|
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive |
|
|
RUN apt-get update && apt-get install -y python3 python3-pip git build-essential cmake wget && rm -rf /var/lib/apt/lists/* |
|
|
|
|
|
|
|
|
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 1 |
|
|
RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1 |
|
|
|
|
|
|
|
|
RUN useradd -m -u 1000 user |
|
|
USER user |
|
|
WORKDIR /home/user/app |
|
|
|
|
|
|
|
|
|
|
|
COPY requirements.txt . |
|
|
ENV CMAKE_ARGS="-DLLAMA_CUBLAS=on" |
|
|
ENV FORCE_CMAKE=1 |
|
|
RUN pip install --no-cache-dir -r requirements.txt |
|
|
|
|
|
|
|
|
COPY app.py . |
|
|
|
|
|
|
|
|
CMD ["python", "app.py"] |