Spaces:
Paused
Paused
Commit
·
5ac6a38
1
Parent(s):
bc1988d
Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -6,8 +6,9 @@ EXPOSE 8000
|
|
| 6 |
|
| 7 |
WORKDIR /app
|
| 8 |
|
|
|
|
| 9 |
RUN wget -qO- "https://cmake.org/files/v3.17/cmake-3.17.0-Linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C /usr/local
|
| 10 |
-
RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install llama-cpp-python
|
| 11 |
COPY requirements.txt ./
|
| 12 |
RUN pip install --upgrade pip && \
|
| 13 |
pip install -r requirements.txt
|
|
|
|
| 6 |
|
| 7 |
WORKDIR /app
|
| 8 |
|
| 9 |
+
RUN export PATH="/usr/local/cuda/bin:$PATH"
|
| 10 |
RUN wget -qO- "https://cmake.org/files/v3.17/cmake-3.17.0-Linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C /usr/local
|
| 11 |
+
RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install llama-cpp-python --no-cache-dir
|
| 12 |
COPY requirements.txt ./
|
| 13 |
RUN pip install --upgrade pip && \
|
| 14 |
pip install -r requirements.txt
|