Skip to content
Snippets Groups Projects
Commit f4a62a0d authored by Ujwal Kundur's avatar Ujwal Kundur
Browse files

Add CUDA Post-Install commands

parent 465932e1
No related branches found
No related tags found
1 merge request!338Base oneapi_jug_xl off jug_xl
This commit is part of merge request !338. Comments created here will be created in the context of that merge request.
...@@ -326,17 +326,24 @@ RUN --mount=type=cache,target=/var/cache/apt \ ...@@ -326,17 +326,24 @@ RUN --mount=type=cache,target=/var/cache/apt \
cuda-cudart-11-7 \ cuda-cudart-11-7 \
cuda-compat-11-7 \ cuda-compat-11-7 \
&& apt-get -yqq autoremove \ && apt-get -yqq autoremove \
<<<<<<< HEAD
<<<<<<< HEAD <<<<<<< HEAD
&& ln -s cuda-11.7 /usr/local/cuda \ && ln -s cuda-11.7 /usr/local/cuda \
=======
&& ln -s cuda-11.7 /usr/local/cuda && \
>>>>>>> Add CUDA Post-Install commands
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
<<<<<<< HEAD
======= =======
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
>>>>>>> Add CUDA libraries >>>>>>> Add CUDA libraries
=======
>>>>>>> Add CUDA Post-Install commands
## copy over everything we need from staging in a single layer ## copy over everything we need from staging in a single layer
RUN --mount=from=staging,target=/staging \ RUN --mount=from=staging,target=/staging \
rm -rf /usr/local \ rm -rf /usr/local \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment