Skip to content

Commit 4a7ed14

Browse files
committed
[Manywheel] Add /usr/local/cuda symlink
And add `nvcc` to path Regression introduced by #1447 when NVIDIA image was dropped in favor of base `centos` image
1 parent bcd1f7b commit 4a7ed14

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

manywheel/Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,8 @@ FROM cpu_final as cuda_final
154154
RUN rm -rf /usr/local/cuda-${BASE_CUDA_VERSION}
155155
COPY --from=cuda /usr/local/cuda-${BASE_CUDA_VERSION} /usr/local/cuda-${BASE_CUDA_VERSION}
156156
COPY --from=magma /usr/local/cuda-${BASE_CUDA_VERSION} /usr/local/cuda-${BASE_CUDA_VERSION}
157+
RUN ln -sf /usr/local/cuda-${BASE_CUDA_VERSION} /usr/local/cuda
158+
ENV PATH=/usr/local/cuda/bin:$PATH
157159

158160
FROM cpu_final as rocm_final
159161
ARG ROCM_VERSION=3.7

0 commit comments

Comments
 (0)