updated setup.sh

2026-07-02 05:51:44 +00:00
parent d78acd3ba7
commit 5009183c67
3 changed files with 17 additions and 3 deletions
@@ -0,0 +1,4 @@
+import tensorflow as tf
+
+print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
+print("Is built with CUDA: ", tf.test.is_built_with_cuda())