Spaces:
Running
on
T4
Running
on
T4
Update Dockerfile
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -16,6 +16,7 @@ WORKDIR /app
|
|
| 16 |
COPY ./requirements.txt /app/requirements.txt
|
| 17 |
|
| 18 |
RUN pip3 install --no-cache-dir --upgrade -r /app/requirements.txt
|
|
|
|
| 19 |
|
| 20 |
|
| 21 |
RUN useradd -m -u 1000 user
|
|
|
|
| 16 |
COPY ./requirements.txt /app/requirements.txt
|
| 17 |
|
| 18 |
RUN pip3 install --no-cache-dir --upgrade -r /app/requirements.txt
|
| 19 |
+
RUN CMAKE_ARGS="-DLLAMA_CUDA=on" pip3 install --no-cache-dir llama-cpp-python
|
| 20 |
|
| 21 |
|
| 22 |
RUN useradd -m -u 1000 user
|