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