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

Add CUDA Post-Install commands

parent 58fb0f93
No related branches found
No related tags found
No related merge requests found
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 \
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