Skip to content

TPU profiling extension#5

Open
ketyi wants to merge 3 commits intotaomanwai:masterfrom
ketyi:master
Open

TPU profiling extension#5
ketyi wants to merge 3 commits intotaomanwai:masterfrom
ketyi:master

Conversation

@ketyi
Copy link

@ketyi ketyi commented Nov 28, 2018

TensorBoard startup is extended with TPU profiling option (in case of TPU is present in the Colab environment)

See: https://cloud.google.com/tpu/docs/cloud-tpu-tools

Copy link
Owner

@taomanwai taomanwai left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the contribution!

But when I tried, error "No OpKernel was registered to support Op 'ConfigureDistributedTPU' with these attrs" occurs!

Do u know why? Or do u have full set of testing code which works? So that i can debug too!

To avoid the "No OpKernel was registered to support Op 'ConfigureDistributedTPU' with these attrs"
To avoid the "No OpKernel was registered to support Op 'ConfigureDistributedTPU' with these attrs"
@ketyi
Copy link
Author

ketyi commented Dec 30, 2018

I think this TPU error has nothing to do with my modifications. In my experience it can be due to compatibility issues between the installed Tensorflow and Keras versions. A solution is to import the Keras pieces from tensorflow.python.keras . I have modified it accordingly, could you give it a try?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants