RUBY / sample.py
itzbhav's picture
Upload 8 files
748b6ee verified
raw
history blame contribute delete
187 Bytes
import tensorflow as tf
# Check if GPU is available
if tf.test.is_gpu_available():
print("GPU is available")
else:
print("GPU is not available")
# Rest of your code...