Spaces:

davanstrien
/

synthetic-data-workshop

Runtime error

davanstrien HF Staff commited on Jun 19, 2024

Commit

f0c2c24

1 Parent(s): 9edb8b6

Update GPU detection method in check_gpu.sh

Files changed (1) hide show

check_gpu.sh CHANGED Viewed

@@ -1,6 +1,6 @@
 #!/bin/bash
-if lspci | grep -i nvidia; then
     exit 0
 else
     exit 1

 #!/bin/bash
+if command -v nvidia-smi &> /dev/null && nvidia-smi -L; then
     exit 0
 else
     exit 1