We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
/usr/local/cuda
1 parent bcd1f7b commit 4a7ed14Copy full SHA for 4a7ed14
manywheel/Dockerfile
@@ -154,6 +154,8 @@ FROM cpu_final as cuda_final
154
RUN rm -rf /usr/local/cuda-${BASE_CUDA_VERSION}
155
COPY --from=cuda /usr/local/cuda-${BASE_CUDA_VERSION} /usr/local/cuda-${BASE_CUDA_VERSION}
156
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
159
160
FROM cpu_final as rocm_final
161
ARG ROCM_VERSION=3.7
0 commit comments