Skip to content

GitLab

Commit 5a81390b authored by Michael Yang's avatar Michael Yang
Browse files

update dockerfile.cuda

parent a89ef99a
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
Dockerfile.cuda
View file @ 5a81390b
FROM nvidia/cuda:12.2.0-devel-ubuntu22.04 ARG CUDA_VERSION=12.2.0
FROM nvidia/cuda:$CUDA_VERSION-devel-ubuntu22.04
ARG TARGETARCH ARG TARGETARCH
ARG VERSION=0.0.0 ARG VERSION=0.0.0
...@@ -13,9 +15,11 @@ ENV GOARCH=$TARGETARCH ...@@ -13,9 +15,11 @@ ENV GOARCH=$TARGETARCH
RUN /usr/local/go/bin/go generate ./... \ RUN /usr/local/go/bin/go generate ./... \
&& /usr/local/go/bin/go build -ldflags "-linkmode=external -extldflags='-static' -X=github.com/jmorganca/ollama/version.Version=$VERSION -X=github.com/jmorganca/ollama/server.mode=release" . && /usr/local/go/bin/go build -ldflags "-linkmode=external -extldflags='-static' -X=github.com/jmorganca/ollama/version.Version=$VERSION -X=github.com/jmorganca/ollama/server.mode=release" .
FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04 FROM ubuntu:22.04
ENV OLLAMA_HOST 0.0.0.0 ENV OLLAMA_HOST 0.0.0.0
RUN apt-get update && apt-get install -y ca-certificates
ARG USER=ollama ARG USER=ollama
ARG GROUP=ollama ARG GROUP=ollama
RUN groupadd $GROUP && useradd -m -g $GROUP $USER RUN groupadd $GROUP && useradd -m -g $GROUP $USER
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment