Skip to content

Commit

Permalink
Merge pull request #182 from AdamBrousseau/cuda_unversioned
Browse files Browse the repository at this point in the history
Don't version the cuda install folder
  • Loading branch information
AdamBrousseau authored Sep 22, 2023
2 parents b0e682a + a1d7216 commit e1d7396
Showing 1 changed file with 4 additions and 7 deletions.
11 changes: 4 additions & 7 deletions pipelines/build/dockerFiles/cuda.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,10 @@ ARG cuda_distro=ubuntu20.04
FROM nvidia/cuda:${cuda_ver}-devel-${cuda_distro} as cuda
FROM $image

# This line should be present here to let access to cuda_ver ARG in the following lines
ARG cuda_ver

# Install cuda headers https://github.com/eclipse/openj9/blob/master/buildenv/docker/mkdocker.sh#L586-L593
RUN mkdir -p /usr/local/cuda-$cuda_ver/nvvm
RUN mkdir -p /usr/local/cuda/nvvm
# TEMP: Copy the header files from the local compile box to the container.
COPY --from=cuda /usr/local/cuda/include /usr/local/cuda-${cuda_ver}/include
COPY --from=cuda /usr/local/cuda/nvvm/include /usr/local/cuda-${cuda_ver}/nvvm/include
COPY --from=cuda /usr/local/cuda/include /usr/local/cuda/include
COPY --from=cuda /usr/local/cuda/nvvm/include /usr/local/cuda/nvvm/include

ENV CUDA_HOME="/usr/local/cuda-${cuda_ver}"
ENV CUDA_HOME="/usr/local/cuda"

0 comments on commit e1d7396

Please sign in to comment.