Skip to content

Commit 8583ea5

Browse files
committed
add nccl build for archs 80,90
1 parent 709563e commit 8583ea5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

dockerfile/cuda12.8.dockerfile

+4-1
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,10 @@ RUN if [ "$TARGETARCH" = "amd64" ]; then \
126126
RUN cd /tmp && \
127127
git clone -b v2.25.1-1 https://github.com/NVIDIA/nccl.git && \
128128
cd nccl && \
129-
make -j ${NUM_MAKE_JOBS} src.build NVCC_GENCODE="-gencode=arch=compute_100,code=sm_100" && \
129+
make -j ${NUM_MAKE_JOBS} src.build \
130+
NVCC_GENCODE="-gencode=arch=compute_100,code=sm_100 \
131+
-gencode=arch=compute_100,code=sm_100 \
132+
-gencode=arch=compute_100,code=sm_100" && \
130133
make install && \
131134
rm -rf /tmp/nccl
132135

0 commit comments

Comments
 (0)