Spaces:

DSatishchandra
/

Solar_Panels

Runtime error

DSatishchandra commited on May 21

Commit

fb3320e

verified ·

1 Parent(s): ed32826

Update train_lstm_model.py

Files changed (1) hide show

train_lstm_model.py CHANGED Viewed

@@ -9,7 +9,7 @@ import shutil
 import tensorflow as tf
 # Disable GPU to avoid cuDNN/cuFFT/cuBLAS conflicts
-os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Force TensorFlow to use CPU
 print("CUDA_VISIBLE_DEVICES set to empty to disable GPU usage.")
 # Check TensorFlow version

 import tensorflow as tf
 # Disable GPU to avoid cuDNN/cuFFT/cuBLAS conflicts
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
 print("CUDA_VISIBLE_DEVICES set to empty to disable GPU usage.")
 # Check TensorFlow version