Skip to content

Instantly share code, notes, and snippets.

@b0noI

b0noI/tpu-test.py

Created August 8, 2018 17:38
Show Gist options
  • Save b0noI/a13f3a5be985882a68b1f69458180e9c to your computer and use it in GitHub Desktop.
Save b0noI/a13f3a5be985882a68b1f69458180e9c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tpu-test.py
import tensorflow as tf
TPU_WORKER = '10.0.101.2:8470' # put your IP!
session = tf.Session('grpc://' + TPU_WORKER,
config=tf.ConfigProto(isolate_session_state=True))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment