| |
| |
| 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"] |