Skip to content

Commit cdb209a

Browse files
committed
print cuda node info
1 parent 7e9a65c commit cdb209a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

.github/scripts/build-cuda.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ for NO_CUBLASLT in ON OFF; do
1515
docker run --platform "linux/$build_arch" -i -w /src -v "$PWD:/src" "$image" sh -c \
1616
"apt-get update \
1717
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
18-
&& cmake -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" -DNO_CUBLASLT=${NO_CUBLASLT} . \
18+
&& nvidia-smi && cmake -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" -DNO_CUBLASLT=${NO_CUBLASLT} . \
1919
&& cmake --build ."
2020
else
2121
pip install cmake==3.28.3

0 commit comments

Comments
 (0)