Search Unity

  1. Welcome to the Unity Forums! Please take the time to read our Code of Conduct to familiarize yourself with the forum rules and how to post constructively.

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?
     
    DeshanGunarathna, Jan 12, 2023
    #1
    shivamdude likes this.