Update Dockerfile
Browse files- Dockerfile +7 -2
Dockerfile
CHANGED
|
@@ -27,8 +27,13 @@ RUN python3 -m pip install --upgrade pip pytest cmake \
|
|
| 27 |
# Install llama-cpp-python (build with cuda)
|
| 28 |
# RUN CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=75" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python --force-reinstall --upgrade --no-cache-dir --verbose
|
| 29 |
# RUN python3 -m pip install llama-cpp-python
|
| 30 |
-
RUN python3 -m pip install llama-cpp-python \
|
| 31 |
-
--extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
RUN useradd -m -u 1000 user
|
| 34 |
# Switch to the "user" user
|
|
|
|
| 27 |
# Install llama-cpp-python (build with cuda)
|
| 28 |
# RUN CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=75" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python --force-reinstall --upgrade --no-cache-dir --verbose
|
| 29 |
# RUN python3 -m pip install llama-cpp-python
|
| 30 |
+
#RUN python3 -m pip install llama-cpp-python \
|
| 31 |
+
# --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
|
| 32 |
+
|
| 33 |
+
RUN CMAKE_ARGS="-DGGML_CUDA=on" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python \
|
| 34 |
+
--extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121 \
|
| 35 |
+
--force-reinstall --no-cache-dir --verbose
|
| 36 |
+
|
| 37 |
|
| 38 |
RUN useradd -m -u 1000 user
|
| 39 |
# Switch to the "user" user
|