37 lines
1.6 KiB
Text
37 lines
1.6 KiB
Text
# amd64 dependencies
|
|
FROM --platform=linux/amd64 nvidia/cuda:11.3.1-devel-centos7 AS base-amd64
|
|
RUN yum install -y https://repo.ius.io/ius-release-el7.rpm centos-release-scl && \
|
|
yum update -y && \
|
|
yum install -y devtoolset-10-gcc devtoolset-10-gcc-c++ git236 wget
|
|
RUN wget "https://github.com/Kitware/CMake/releases/download/v3.27.6/cmake-3.27.6-linux-x86_64.sh" -O cmake-installer.sh && chmod +x cmake-installer.sh && ./cmake-installer.sh --skip-license --prefix=/usr/local
|
|
ENV PATH /opt/rh/devtoolset-10/root/usr/bin:$PATH
|
|
|
|
# arm64 dependencies
|
|
FROM --platform=linux/arm64 nvidia/cuda:11.3.1-devel-ubuntu18.04 AS base-arm64
|
|
RUN apt-get update && \
|
|
apt-get install -y git build-essential wget software-properties-common && \
|
|
add-apt-repository -y ppa:ubuntu-toolchain-r/test && \
|
|
apt-get update && \
|
|
apt-get install -y gcc-9 g++-9 && \
|
|
update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 60 --slave /usr/bin/g++ g++ /usr/bin/g++-9 && \
|
|
update-alternatives --config gcc
|
|
RUN wget "https://github.com/Kitware/CMake/releases/download/v3.27.6/cmake-3.27.6-linux-aarch64.sh" -O cmake-installer.sh && chmod +x cmake-installer.sh && ./cmake-installer.sh --skip-license --prefix=/usr/local
|
|
|
|
FROM base-${TARGETARCH}
|
|
ARG TARGETARCH
|
|
ARG GOFLAGS="'-ldflags -w -s'"
|
|
|
|
# install go
|
|
ADD https://dl.google.com/go/go1.21.3.linux-$TARGETARCH.tar.gz /tmp/go1.21.3.tar.gz
|
|
RUN mkdir -p /usr/local && tar xz -C /usr/local </tmp/go1.21.3.tar.gz
|
|
|
|
# build the final binary
|
|
WORKDIR /go/src/github.com/jmorganca/ollama
|
|
COPY . .
|
|
|
|
ENV GOOS=linux
|
|
ENV GOARCH=$TARGETARCH
|
|
ENV GOFLAGS=$GOFLAGS
|
|
|
|
RUN /usr/local/go/bin/go generate ./... && \
|
|
/usr/local/go/bin/go build .
|