File tree Expand file tree Collapse file tree 2 files changed +20
-2
lines changed Expand file tree Collapse file tree 2 files changed +20
-2
lines changed Original file line number Diff line number Diff line change @@ -17,7 +17,25 @@ ARG CUDA_VERSION
1717ARG GOLANG_VERSION=x.x.x
1818ARG VERSION="N/A"
1919
20- FROM golang:${GOLANG_VERSION} AS build
20+ FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-${BASE_DIST} as build
21+
22+ RUN yum install -y wget make git gcc
23+
24+ ARG GOLANG_VERSION=0.0.0
25+ RUN set -eux; \
26+ \
27+ arch="$(uname -m)"; \
28+ case "${arch##*-}" in \
29+ x86_64 | amd64) ARCH='amd64' ;; \
30+ ppc64el | ppc64le) ARCH='ppc64le' ;; \
31+ aarch64) ARCH='arm64' ;; \
32+ *) echo "unsupported architecture" ; exit 1 ;; \
33+ esac; \
34+ wget -nv -O - https://storage.googleapis.com/golang/go${GOLANG_VERSION}.linux-${ARCH}.tar.gz \
35+ | tar -C /usr/local -xz
36+
37+ ENV GOPATH /go
38+ ENV PATH $GOPATH/bin:/usr/local/go/bin:$PATH
2139
2240WORKDIR /build
2341COPY . .
Original file line number Diff line number Diff line change @@ -18,4 +18,4 @@ vVERSION := v$(VERSION:v%=%)
1818
1919CUDA_VERSION := 12.2.0
2020
21- GOLANG_VERSION := 1.20
21+ GOLANG_VERSION := 1.20.4
You can’t perform that action at this time.
0 commit comments