Search Unity

  1. Unity support for visionOS is now available. Learn more in our blog post.
    Dismiss Notice

Affordable and Optimal Hardware for Unity ML Agents Training

Discussion in 'ML-Agents' started by hendrikodito, May 19, 2020.

  1. hendrikodito


    Mar 6, 2019
    Hi creators,

    Thank you so much for creating this amazing tool for us to use. I am currently enrolled in a master of Architecture program and I have decided to use ML-Agents as my main driver for the project. With this new devotion, I am expecting to upgrade my machine.

    I have read that ML-Agents do not require GPU even training could be done by only the CPU. I have some questions regarding this:
    1. Would a better GPU in ML-Agents than running it on the CPU? I am expecting myself to upgrade to NVDA RTX 2070 Super with AMD Ryzen 5 3600/
    2. Since the training option is binary: GPU or CPU, would that mean I could just invest in one CPU better than having a better GPU?
    3. What would you suggest for me to upgrade, or, maybe for the good for all, you could suggest any specific combination of hardware for a range of budget?

    I am currently using a laptop with Intel 6700HQ and NVDA GTX 960 M and all my simple agent training ran pretty well so far.

    Thank you so much!
  2. mbaske


    Dec 31, 2017
    Last edited: Jun 17, 2020
  3. HanAusBerlin


    Sep 13, 2017
    I set up ML agents with GPU (1080 gtx) with almost no effect. GPU support seems to be poor at this point.