forked from livepeer/go-livepeer
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Bump nvidia cuda docker image version (livepeer#2821)
Co-authored-by: hjpotter92 <[email protected]>
- Loading branch information
1 parent
84554cc
commit a708fda
Showing
9 changed files
with
36 additions
and
34 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,4 +1,4 @@ | ||
FROM --platform=$BUILDPLATFORM ubuntu:18.04 as build | ||
FROM --platform=$BUILDPLATFORM ubuntu:20.04 as build | ||
|
||
ARG TARGETARCH | ||
ARG BUILDARCH | ||
|
@@ -30,7 +30,7 @@ RUN GRPC_HEALTH_PROBE_VERSION=v0.3.6 \ | |
&& ldconfig /usr/local/lib | ||
|
||
# note: for runtime, Tensorflow version needs to be compatible with CUDA and CuDNN of the image | ||
RUN LIBTENSORFLOW_VERSION=2.3.4 \ | ||
RUN LIBTENSORFLOW_VERSION=2.12.1 \ | ||
&& curl -LO https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-${LIBTENSORFLOW_VERSION}.tar.gz \ | ||
&& mkdir /tf && tar -C /tf -xzf libtensorflow-gpu-linux-x86_64-${LIBTENSORFLOW_VERSION}.tar.gz | ||
|
||
|
@@ -46,32 +46,32 @@ RUN mkdir -p /go \ | |
COPY ./install_ffmpeg.sh ./install_ffmpeg.sh | ||
|
||
ARG BUILD_TAGS | ||
ENV BUILD_TAGS=${BUILD_TAGS} | ||
ENV BUILD_TAGS=${BUILD_TAGS} | ||
|
||
COPY go.mod go.sum ./ | ||
RUN go mod download | ||
|
||
RUN ./install_ffmpeg.sh \ | ||
RUN ./install_ffmpeg.sh \ | ||
&& GO111MODULE=on go get -v github.com/golangci/golangci-lint/cmd/[email protected] \ | ||
&& go get -v github.com/jstemmer/go-junit-report | ||
|
||
COPY . . | ||
|
||
RUN make livepeer livepeer_cli livepeer_bench livepeer_router | ||
|
||
# cuda 10.2 image is 1.4 Gb smaller, which is critical for our root partition size | ||
FROM --platform=$TARGETPLATFORM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 AS livepeer-amd64-base | ||
FROM --platform=$TARGETPLATFORM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04 AS livepeer-amd64-base | ||
|
||
FROM --platform=$TARGETPLATFORM nvidia/cuda:11.6.2-cudnn8-runtime-ubuntu20.04 AS livepeer-arm64-base | ||
FROM --platform=$TARGETPLATFORM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04 AS livepeer-arm64-base | ||
|
||
FROM livepeer-${TARGETARCH}-base | ||
FROM livepeer-${TARGETARCH}-base | ||
|
||
ENV NVIDIA_DRIVER_CAPABILITIES=all | ||
|
||
COPY --from=build /build/ /usr/local/bin/ | ||
COPY --from=build /usr/bin/grpc_health_probe /usr/local/bin/grpc_health_probe | ||
COPY --from=build /src/tasmodel.pb /tasmodel.pb | ||
COPY --from=build /usr/share/misc/pci.ids /usr/share/misc/pci.ids | ||
|
||
# libtensorflow.so is required at runtime, because Ffmpeg DNN filter loads it dynamically | ||
COPY --from=build /tf/ /usr/local/ | ||
|
||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters