Updated dockerfiles

Author: djthorpe

Dockerfile.cuda

@@ -44,7 +44,7 @@ ARG ARCH
 ARG CUDA
 
 RUN apt-get update \
-  && apt-get -y install ca-certificates gnupg wget libgomp1 \
+  && apt-get -y install ca-certificates gnupg wget \
   && wget https://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}/${ARCH}/cuda-keyring_1.1-1_all.deb \
   && dpkg -i cuda-keyring_1.1-1_all.deb \
   && rm cuda-keyring_1.1-1_all.deb \

Dockerfile.llamacpp

@@ -21,7 +21,7 @@ ARG ARCH
 ARG BASE_IMAGE_BUILD
 ARG BASE_IMAGE_RUNTIME
 
-RUN apt-get -y install libcurl4 curl
+RUN apt-get -y install libcurl4 curl libgomp1
 COPY --from=build /app/llama.cpp/llama-server /usr/local/bin/llama-server
 HEALTHCHECK CMD [ "curl", "-f", "http://localhost:8080/health" ]
 ENTRYPOINT [ "/usr/local/bin/llama-server" ]