vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

pip later

Browse source
This commit is contained in:
Robert Washbourne 2023-11-23 23:43:23 -05:00
parent 93f86c98d1
commit 4b6e344bad
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
.devops/main-cuda.Dockerfile
View file

@ -13,7 +13,7 @@ ARG CUDA_DOCKER_ARCH=all
# ARG CUDA_DOCKER_ARCH=sm_86 # ARG CUDA_DOCKER_ARCH=sm_86
RUN apt-get update && \ RUN apt-get update && \
apt-get install -y build-essential git wget python3 python3-pip apt-get install -y build-essential git wget
WORKDIR /app WORKDIR /app
@ -37,6 +37,7 @@ ENV MODEL_URL=${MODEL_URL}
RUN wget $MODEL_URL -O /model.gguf RUN wget $MODEL_URL -O /model.gguf
WORKDIR /install WORKDIR /install
RUN apt-get install python3 python3-pip
RUN pip install --install-option="--prefix=/install" runpod RUN pip install --install-option="--prefix=/install" runpod
FROM ${BASE_CUDA_RUN_CONTAINER} as runtime FROM ${BASE_CUDA_RUN_CONTAINER} as runtime