[Refactor] ♻️ Docker Build Template #3

Author: Harry-zklcdc

Dockerfile

@@ -1,12 +1,12 @@
-FROM golang:1.21.4-alpine AS builder
+FROM golang:1.23.6-alpine AS builder
 LABEL stage=gobuilder \
-      mainatiner=https://github.com/mayooot/gpu-docker-api
+      mainatiner=https://github.com/XShengTech/gpu-docker-api
 
-RUN sed -i 's/dl-cdn.alpinelinux.org/mirrors.tuna.tsinghua.edu.cn/g' /etc/apk/repositories
-RUN apk add gcc g++ make libffi-dev openssl-dev libtool
+# RUN sed -i 's/dl-cdn.alpinelinux.org/mirrors.tuna.tsinghua.edu.cn/g' /etc/apk/repositories
+RUN apk add gcc g++ make libffi-dev openssl-dev libtool git
 
 ENV CGO_ENABLED 0
-ENV GOPROXY https://goproxy.cn,direct
+# ENV GOPROXY https://goproxy.cn,direct
 
 WORKDIR /build
 
@@ -15,14 +15,19 @@ COPY go.sum .
 RUN go mod download
 
 COPY . .
-RUN make linux_no_ldflags
+RUN BRANCH=$(git rev-parse --abbrev-ref HEAD) && \
+    VERSION=$(git describe --tags | cut -d'-' -f1) && \
+    COMMIT=$(git rev-parse --short HEAD) && \
+    GO_VERSION=$(go env GOVERSION) && \
+    BUILD_TIME=$(date +%FT%T%z) && \
+    go build -ldflags="-s -w -X main.BRANCH=${BRANCH} -X main.VERSION=${VERSION} -X main.COMMIT=${COMMIT} -X main.GoVersion=${GO_VERSION} -X main.BuildTime=${BUILD_TIME}" -trimpath -o gpu-docker-api cmd/gpu-docker-api/main.go
 
-FROM nvidia/cuda:10.0-base
+FROM ubuntu:22.04
 
 VOLUME /data
 WORKDIR /data
 
-COPY --from=builder /build/bin/gpu-docker-api-linux-amd64 /data/gpu-docker-api
+COPY --from=builder /build/gpu-docker-api /data/gpu-docker-api
 
 EXPOSE 2378
 

docker-compose.yaml

@@ -35,10 +35,13 @@ services:
 
   gpu-docker-api:
     hostname: gpu-docker-api
-    image: mayooot/gpu-docker-api:v0.0.3
+    image: xshengtech/gpu-docker-api:latest
     container_name: gpu-docker-api
     restart: unless-stopped
     network_mode: host
+    runtime: nvidia
+    environment:
+      - APIKEY=CHANGETHIS
     deploy:
       resources:
         reservations: