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.

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