From 7ca6fd8af6ff3e76da55ddf2b28bcbbdd69b8f21 Mon Sep 17 00:00:00 2001
From: Ujwal Kundur <ujwal.kundur@gmail.com>
Date: Thu, 18 Aug 2022 12:06:23 +0530
Subject: [PATCH] Add CUDA Post-Install commands
---
containers/oneapi/onedev.Dockerfile | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/containers/oneapi/onedev.Dockerfile b/containers/oneapi/onedev.Dockerfile
index 382de16ef..b113a02e9 100644
--- a/containers/oneapi/onedev.Dockerfile
+++ b/containers/oneapi/onedev.Dockerfile
@@ -322,8 +322,12 @@ RUN --mount=type=cache,target=/var/cache/apt \
cuda-cudart-11-7 \
cuda-compat-11-7 \
&& apt-get -yqq autoremove \
+ && ln -s cuda-11.7 /usr/local/cuda && \
&& rm -rf /var/lib/apt/lists/*
+ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
+ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
+
## copy over everything we need from staging in a single layer
RUN --mount=from=staging,target=/staging \
rm -rf /usr/local \
--
GitLab