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.
1 parent 1b1e7cd commit 4f43daeCopy full SHA for 4f43dae
docker/Dockerfile.rocm_base
@@ -18,7 +18,7 @@ FROM ${BASE_IMAGE} AS base
18
ENV PATH=/opt/rocm/llvm/bin:$PATH
19
ENV ROCM_PATH=/opt/rocm
20
ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:
21
-ARG PYTORCH_ROCM_ARCH=gfx942;gfx950;
+ARG PYTORCH_ROCM_ARCH=gfx942;gfx950
22
ENV PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH}
23
# ENV AITER_ROCM_ARCH=gfx942;gfx950
24
0 commit comments