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 7e9a65c commit cdb209aCopy full SHA for cdb209a
1 file changed
.github/scripts/build-cuda.sh
@@ -15,7 +15,7 @@ for NO_CUBLASLT in ON OFF; do
15
docker run --platform "linux/$build_arch" -i -w /src -v "$PWD:/src" "$image" sh -c \
16
"apt-get update \
17
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
18
- && 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} . \
19
&& cmake --build ."
20
else
21
pip install cmake==3.28.3
0 commit comments