print cuda node info

bitsandbytes-foundation · Jul 30, 2024 · cdb209a · cdb209a
1 parent 7e9a65c
commit cdb209a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/scripts/build-cuda.sh b/.github/scripts/build-cuda.sh
@@ -15,7 +15,7 @@ for NO_CUBLASLT in ON OFF; do
 		docker run --platform "linux/$build_arch" -i -w /src -v "$PWD:/src" "$image" sh -c \
 			"apt-get update \
       && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
-      && cmake -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" -DNO_CUBLASLT=${NO_CUBLASLT} . \
+      && nvidia-smi && cmake -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" -DNO_CUBLASLT=${NO_CUBLASLT} . \
       && cmake --build ."
 	else
 		pip install cmake==3.28.3