Search code examples
google-ai-platform

Does TPU Accelerators are supported in Unified AI Platform

I have some custom training jobs in AI Platform using GPUs & TPUs and trying to run the same jobs in unified AI platform but not able to find the TPU machines.

Solution

    • Looking into the AI Platform Unified documentation able to find GPUs configurations but nothing about TPUs. Looks like TPU’s are not supported by unified.

    • You can look into the AI Platform Unified release notes updates to check update about this feature