Search Unity

Resolved GPU Training using ML-Agents (CUDA & cuDNN)

Discussion in 'ML-Agents' started by mateolopezareal, Jun 12, 2020.

  1. mateolopezareal

    mateolopezareal

    Joined:
    Jun 4, 2020
    Posts:
    54
    I am following the tutorial to install ML-Agents in my windows laptop, and I am not sure what version of CUDA and cuDNN should I install, can I install the newest?
     
    mateolopezareal, Jun 12, 2020
    #1

  2. MrWetsnow

    MrWetsnow

    Joined:
    Jan 5, 2020
    Posts:
    60
    MrWetsnow, Jun 12, 2020
    #2

  3. vincentgao88

    vincentgao88

    Unity Technologies

    Joined:
    Feb 7, 2018
    Posts:
    21
    +1 @MrWetsnow , the version of CUDA depends on your tensorflow version.
     
    vincentgao88, Jun 15, 2020
    #3