Skip to content

Commit

Permalink
🔨 Updated ubuntu2004_gpu.dockerfile to avoid deprecated missing base …
Browse files Browse the repository at this point in the history
…image as well as add instructions to COPY necessary bash scripts for installation of dependencies.

Signed-off-by: Bey Hao Yun <[email protected]>
  • Loading branch information
cardboardcode committed Sep 21, 2024
1 parent f3f40a0 commit 6535836
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion dockerfiles/ubuntu2004_gpu.dockerfile
Original file line number Diff line number Diff line change
@@ -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 \
Expand Down

0 comments on commit 6535836

Please sign in to comment.