diff --git a/containers/oneapi/onedev.Dockerfile b/containers/oneapi/onedev.Dockerfile index 382de16ef7d091e407092c9a14963055423e460f..b113a02e9ccf1e2a6978cc21ff481e2dde5c58a3 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 \