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
......@@ -326,17 +326,24 @@ RUN --mount=type=cache,target=/var/cache/apt \
cuda-cudart-11-7 \
cuda-compat-11-7 \
&& apt-get -yqq autoremove \
<<<<<<< HEAD
<<<<<<< HEAD
&& 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/*
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
<<<<<<< HEAD
=======
&& rm -rf /var/lib/apt/lists/*
>>>>>>> Add CUDA libraries
=======
>>>>>>> Add CUDA Post-Install commands
## copy over everything we need from staging in a single layer
RUN --mount=from=staging,target=/staging \
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