ankurgup510
commited on
Commit
•
ab29088
1
Parent(s):
59a00b4
Create melt.py
Browse files
melt.py
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import numpy as np
|
2 |
+
|
3 |
+
import tensorflow as tf
|
4 |
+
import tensorflow_hub as hub
|
5 |
+
import tensorflow_datasets as tfds
|
6 |
+
|
7 |
+
import matplotlib.pyplot as plt
|
8 |
+
|
9 |
+
print("Version: ", tf.__version__)
|
10 |
+
print("Eager mode: ", tf.executing_eagerly())
|
11 |
+
print("Hub version: ", hub.__version__)
|
12 |
+
print("GPU is", "available" if tf.config.list_physical_devices('GPU') else "NOT AVAILABLE")
|