2 ms·
Are TPUs drop-in replacement for CUDA if you were using TF?Can I simply change device from CUDA to TPU and run any TF code? Last I heard, TPUs still had long wa
by sytelus 8y ago
Are TPUs drop-in replacement for CUDA if you were using TF?Can I simply change device from CUDA to TPU and run any TF code? Last I heard, TPUs still had long way to go towards making this happen...
- nl 8y agoIf you are using TF, you can look in Tensorboard at your model graph and it will show you any incompatible operations. These days it's pretty good. Not perfect, but you can do RNNs on it now. The FAQ has decent docs: https://cloud.google.com/tpu/docs/faq https://cloud.google.com/tpu/docs/faq