Skip to content

Commit

Permalink
Don't version the cuda install folder
Browse files Browse the repository at this point in the history
Related ibmruntimes/temurin-build#94

Signed-off-by: Adam Brousseau <adam.brousseau88@gmail.com>
  • Loading branch information
AdamBrousseau committed Sep 22, 2023
1 parent fe2d97d commit a1d7216
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 a1d7216

Please sign in to comment.