test-nvidia/test-nvidia
2019-10-15 12:46:51 +02:00

13 lines
243 B
Python
Executable file

#!/usr/bin/python3
import os
import tensorflow as tf
os.system('nvidia-smi')
hello = tf.constant('Hello, TensorFlow!')
sess = tf.compat.v1.Session()
print(sess.run(hello))
print('GPU available:', tf.test.is_gpu_available(cuda_only=True))