Search Unity

  1. Megacity Metro Demo now available. Download now.
    Dismiss Notice
  2. Unity support for visionOS is now available. Learn more in our blog post.
    Dismiss Notice

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.