File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -70,7 +70,7 @@ RUN ln -sf /usr/local/cuda-12.4 /usr/local/cuda
70
70
FROM cpu as rocm
71
71
ARG PYTORCH_ROCM_ARCH
72
72
ENV PYTORCH_ROCM_ARCH ${PYTORCH_ROCM_ARCH}
73
- # Adding ROCM_PATH env var so that LoadHip.cmake (even with logic updated for ROCm6.0)
73
+ # Adding ROCM_PATH env var so that LoadHip.cmake (even with logic updated for ROCm6.0)
74
74
# find HIP works for ROCm5.7. Not needed for ROCm6.0 and above.
75
75
# Remove below when ROCm5.7 is not in support matrix anymore.
76
76
ENV ROCM_PATH /opt/rocm
Original file line number Diff line number Diff line change @@ -132,7 +132,7 @@ COPY --from=openssl /opt/openssl /opt/openss
132
132
COPY --from=python /opt/python /opt/python
133
133
COPY --from=python /opt/_internal /opt/_internal
134
134
COPY --from=python /opt/python/cp39-cp39/bin/auditwheel /usr/local/bin/auditwheel
135
- COPY --from=intel ${MKLROOT} ${MKLROOT}
135
+ COPY --from=intel ${MKLROOT} ${MKLROOT}
136
136
COPY --from=patchelf /usr/local/bin/patchelf /usr/local/bin/patchelf
137
137
COPY --from=jni /usr/local/include/jni.h /usr/local/include/jni.h
138
138
COPY --from=libpng /usr/local/bin/png* /usr/local/bin/
You can’t perform that action at this time.
0 commit comments