[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
pytorch · Jul 6, 2023 · 4a7ed14 · 4a7ed14
1 parent bcd1f7b
commit 4a7ed14
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/manywheel/Dockerfile b/manywheel/Dockerfile
@@ -154,6 +154,8 @@ FROM cpu_final as cuda_final
 RUN rm -rf /usr/local/cuda-${BASE_CUDA_VERSION}
 COPY --from=cuda     /usr/local/cuda-${BASE_CUDA_VERSION}  /usr/local/cuda-${BASE_CUDA_VERSION}
 COPY --from=magma    /usr/local/cuda-${BASE_CUDA_VERSION}  /usr/local/cuda-${BASE_CUDA_VERSION}
+RUN ln -sf /usr/local/cuda-${BASE_CUDA_VERSION} /usr/local/cuda
+ENV PATH=/usr/local/cuda/bin:$PATH
 
 FROM cpu_final as rocm_final
 ARG ROCM_VERSION=3.7