diff --git a/libtorch/Dockerfile b/libtorch/Dockerfile index c38b7afe4..aef063b5b 100644 --- a/libtorch/Dockerfile +++ b/libtorch/Dockerfile @@ -70,7 +70,7 @@ RUN ln -sf /usr/local/cuda-12.4 /usr/local/cuda FROM cpu as rocm ARG PYTORCH_ROCM_ARCH ENV PYTORCH_ROCM_ARCH ${PYTORCH_ROCM_ARCH} -# Adding ROCM_PATH env var so that LoadHip.cmake (even with logic updated for ROCm6.0) +# Adding ROCM_PATH env var so that LoadHip.cmake (even with logic updated for ROCm6.0) # find HIP works for ROCm5.7. Not needed for ROCm6.0 and above. # Remove below when ROCm5.7 is not in support matrix anymore. ENV ROCM_PATH /opt/rocm diff --git a/manywheel/Dockerfile b/manywheel/Dockerfile index 02a43e9e0..55cf2609e 100644 --- a/manywheel/Dockerfile +++ b/manywheel/Dockerfile @@ -132,7 +132,7 @@ COPY --from=openssl /opt/openssl /opt/openss COPY --from=python /opt/python /opt/python COPY --from=python /opt/_internal /opt/_internal COPY --from=python /opt/python/cp39-cp39/bin/auditwheel /usr/local/bin/auditwheel -COPY --from=intel ${MKLROOT} ${MKLROOT} +COPY --from=intel ${MKLROOT} ${MKLROOT} COPY --from=patchelf /usr/local/bin/patchelf /usr/local/bin/patchelf COPY --from=jni /usr/local/include/jni.h /usr/local/include/jni.h COPY --from=libpng /usr/local/bin/png* /usr/local/bin/