From 65358365497653794ed163ce89836b98cffd6923 Mon Sep 17 00:00:00 2001 From: Bey Hao Yun Date: Sat, 21 Sep 2024 15:57:00 +0800 Subject: [PATCH] :hammer: Updated ubuntu2004_gpu.dockerfile to avoid deprecated missing base image as well as add instructions to COPY necessary bash scripts for installation of dependencies. Signed-off-by: Bey Hao Yun --- dockerfiles/ubuntu2004_gpu.dockerfile | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/dockerfiles/ubuntu2004_gpu.dockerfile b/dockerfiles/ubuntu2004_gpu.dockerfile index 89ce160..13942b0 100644 --- a/dockerfiles/ubuntu2004_gpu.dockerfile +++ b/dockerfiles/ubuntu2004_gpu.dockerfile @@ -1,8 +1,11 @@ -FROM nvidia/cuda:11.4.0-cudnn8-devel-ubuntu20.04 +FROM nvidia/cuda:11.4.3-cudnn8-devel-ubuntu20.04 ENV DEBIAN_FRONTEND=noninteractive WORKDIR /build +COPY ./scripts/install_latest_cmake.bash install_latest_cmake.bash +COPY ./scripts/install_onnx_runtime.bash install_onnx_runtime.bash +COPY ./scripts/install_apps_dependencies.bash install_apps_dependencies.bash RUN apt-get update && \ apt-get install -y --no-install-recommends \