Search Unity

  1. We are migrating the Unity Forums to Unity Discussions. On July 12, the Unity Forums will become read-only.

    Please, do not make any changes to your username or email addresses at id.unity.com during this transition time.

    It's still possible to reply to existing private message conversations during the migration, but any new replies you post will be missing after the main migration is complete. We'll do our best to migrate these messages in a follow-up step.

    On July 15, Unity Discussions will become read-only until July 18, when the new design and the migrated forum contents will go live.


    Read our full announcement for more information and let us know if you have any questions.

Question ML-Agents on TPUs

Discussion in 'ML-Agents' started by DeshanGunarathna, Jan 12, 2023.

  1. DeshanGunarathna

    DeshanGunarathna

    Joined:
    Jan 19, 2022
    Posts:
    16
    Can we use mlagents models in training or inferencing on TPUs (by using Unity and python virtual environment on training) ?
    Is this possible since mlagents using PyTorch for its ML implementations and PyTorch don't support on TPUs?
     
    shivamdude likes this.