File tree Expand file tree Collapse file tree 4 files changed +14
-15
lines changed Expand file tree Collapse file tree 4 files changed +14
-15
lines changed File renamed without changes.
File renamed without changes.
Original file line number Diff line number Diff line change @@ -35,18 +35,16 @@ RUN set -eux; \
3535 wget -nv -O - https://storage.googleapis.com/golang/go${GOLANG_VERSION}.linux-${ARCH}.tar.gz \
3636 | tar -C /usr/local -xz
3737
38-
3938ENV GOPATH /go
4039ENV PATH $GOPATH/bin:/usr/local/go/bin:$PATH
4140
4241WORKDIR /build
4342COPY . .
4443
45- # NOTE: Until the config utilities are properly integrated into the
46- # nvidia-container-toolkit repository, these are built from the `tools` folder
47- # and not `cmd`.
48- RUN GOPATH=/artifacts go install -ldflags="-s -w -X 'main.Version=${VERSION}'" ./tools/...
49-
44+ RUN mkdir /artifacts
45+ ARG VERSION="N/A"
46+ ARG GIT_COMMIT="unknown"
47+ RUN make PREFIX=/artifacts cmd-nvidia-ctk-installer
5048
5149FROM nvidia/cuda:12.6.3-base-ubi8
5250
@@ -71,7 +69,8 @@ RUN PACKAGE_ARCH=${PACKAGE_ARCH/amd64/x86_64} && PACKAGE_ARCH=${PACKAGE_ARCH/arm
7169
7270WORKDIR /work
7371
74- COPY --from=build /artifacts/bin /work
72+ COPY --from=build /artifacts/nvidia-ctk-installer /work/nvidia-ctk-installer
73+ RUN ln -s nvidia-ctk-installer nvidia-toolkit
7574
7675ENV PATH=/work:$PATH
7776
@@ -85,4 +84,4 @@ LABEL description="See summary"
8584
8685RUN mkdir /licenses && mv /NGC-DL-CONTAINER-LICENSE /licenses/NGC-DL-CONTAINER-LICENSE
8786
88- ENTRYPOINT ["/work/nvidia-toolkit "]
87+ ENTRYPOINT ["/work/nvidia-ctk-installer "]
Original file line number Diff line number Diff line change @@ -41,11 +41,10 @@ ENV PATH $GOPATH/bin:/usr/local/go/bin:$PATH
4141WORKDIR /build
4242COPY . .
4343
44- # NOTE: Until the config utilities are properly integrated into the
45- # nvidia-container-toolkit repository, these are built from the `tools` folder
46- # and not `cmd`.
47- RUN GOPATH=/artifacts go install -ldflags="-s -w -X 'main.Version=${VERSION}'" ./tools/...
48-
44+ RUN mkdir /artifacts
45+ ARG VERSION="N/A"
46+ ARG GIT_COMMIT="unknown"
47+ RUN make PREFIX=/artifacts cmd-nvidia-ctk-installer
4948
5049FROM nvcr.io/nvidia/cuda:12.6.3-base-ubuntu20.04
5150
@@ -80,7 +79,8 @@ RUN dpkg -i \
8079
8180WORKDIR /work
8281
83- COPY --from=build /artifacts/bin /work/
82+ COPY --from=build /artifacts/nvidia-ctk-installer /work/nvidia-ctk-installer
83+ RUN ln -s nvidia-ctk-installer nvidia-toolkit
8484
8585ENV PATH=/work:$PATH
8686
@@ -94,4 +94,4 @@ LABEL description="See summary"
9494
9595RUN mkdir /licenses && mv /NGC-DL-CONTAINER-LICENSE /licenses/NGC-DL-CONTAINER-LICENSE
9696
97- ENTRYPOINT ["/work/nvidia-toolkit "]
97+ ENTRYPOINT ["/work/nvidia-ctk-installer "]
You can’t perform that action at this time.
0 commit comments