Update Dockerfile
Browse files- Dockerfile +5 -0
Dockerfile
CHANGED
|
@@ -6,6 +6,11 @@ COPY ./requirements.txt /code/requirements.txt
|
|
| 6 |
|
| 7 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
# Set up a new user named "user" with user ID 1000
|
| 10 |
RUN useradd -m -u 1000 user
|
| 11 |
|
|
|
|
| 6 |
|
| 7 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 8 |
|
| 9 |
+
ENV CMAKE_ARGS="-DGGML_CUDA=on"
|
| 10 |
+
|
| 11 |
+
# Install llama-cpp-python using pip
|
| 12 |
+
RUN pip install llama-cpp-python
|
| 13 |
+
|
| 14 |
# Set up a new user named "user" with user ID 1000
|
| 15 |
RUN useradd -m -u 1000 user
|
| 16 |
|