diff --git a/ollama37.Dockerfile b/ollama37.Dockerfile index 801b489d..15a2b81a 100644 --- a/ollama37.Dockerfile +++ b/ollama37.Dockerfile @@ -14,10 +14,13 @@ FROM rockylinux/rockylinux:8 RUN dnf -y update # Copy only the built binary and any needed assets from the builder stage -COPY --from=builder /usr/local/src/ollama37/ollama /usr/local/bin/ollama +COPY --from=builder /usr/local/src/ollama37 /usr/local/src/ollama37 COPY --from=builder /usr/local/lib64 /usr/local/lib64 COPY --from=builder /usr/local/cuda-11.4/lib64 /usr/local/cuda-11.4/lib64 +# Create a symbolic link from the built binary to /usr/local/bin for easy access +RUN ln -s /usr/local/src/ollama37/ollama /usr/local/bin/ollama + # Set environment variables ENV LD_LIBRARY_PATH="/usr/local/lib64:/usr/local/cuda-11.4/lib64" ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility