File tree Expand file tree Collapse file tree 3 files changed +6
-2
lines changed Expand file tree Collapse file tree 3 files changed +6
-2
lines changed Original file line number Diff line number Diff line change 99 --build-arg TARGET_ARCH=${TARGET_ARCH} \
1010 --build-arg NVIDIA_IMAGE=${NVIDIA_IMAGE} \
1111 --build-arg DEST_DIR=${DEST_DIR} \
12+ --build-arg VERSION=${VERSION} \
1213 --build-arg GOPROXY=https://goproxy.cn,direct \
1314 . -f=docker/Dockerfile -t ${IMG_TAG}
1415
@@ -19,6 +20,7 @@ dockerwithlib:
1920 --build-arg TARGET_ARCH=${TARGET_ARCH} \
2021 --build-arg NVIDIA_IMAGE=${NVIDIA_IMAGE} \
2122 --build-arg DEST_DIR=${DEST_DIR} \
23+ --build-arg VERSION=${VERSION} \
2224 --build-arg GOPROXY=https://goproxy.cn,direct \
2325 . -f=docker/Dockerfile.withlib -t ${IMG_TAG}
2426
Original file line number Diff line number Diff line change @@ -4,10 +4,11 @@ ARG NVIDIA_IMAGE=nvidia/cuda:12.2.0-devel-ubuntu20.04
44FROM $GOLANG_IMAGE AS build
55FROM $GOLANG_IMAGE AS gobuild
66ARG GOPROXY
7+ ARG VERSION
78ADD . /k8s-vgpu
89# RUN --mount=type=cache,target=/go/pkg/mod \
910# cd /k8s-vgpu && make all
10- RUN cd /k8s-vgpu && make all
11+ RUN cd /k8s-vgpu && make all VERSION=$VERSION
1112RUN go install github.com/NVIDIA/mig-parted/cmd/nvidia-mig-parted@latest
1213
1314FROM $NVIDIA_IMAGE AS nvbuild
Original file line number Diff line number Diff line change @@ -5,8 +5,9 @@ FROM $GOLANG_IMAGE AS build
55FROM $GOLANG_IMAGE AS GOBUILD
66ADD . /k8s-vgpu
77ARG GOPROXY=https://goproxy.cn,direct
8+ ARG VERSION
89RUN go env -w GO111MODULE=on
9- RUN cd /k8s-vgpu && make all
10+ RUN cd /k8s-vgpu && make all VERSION=$VERSION
1011RUN go install github.com/NVIDIA/mig-parted/cmd/nvidia-mig-parted@latest
1112
1213#FROM ubuntu:24.04
You can’t perform that action at this time.
0 commit comments