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.
  2. We have updated the language to the Editor Terms based on feedback from our employees and community. Learn more.
    Dismiss Notice

train with GPU

Discussion in 'ML-Agents' started by kahabal, May 22, 2020.

  1. kahabal

    kahabal

    Joined:
    Nov 19, 2018
    Posts:
    16
    Hi,
    i am using the 1.0 MLAgents and i manage to set up (hypotetically) the tensorflow with the GPU

    tensorflow/core/common_runtime/gpu/gpu_device.cc:1561] Found device 0 with properties:
    pciBusID: 0000:01:00.0 name: GeForce RTX 2070 computeCapability: 7.5
    coreClock: 1.62GHz coreCount: 36 deviceMemorySize: 8.00GiB deviceMemoryBandwidth: 417.29GiB/s

    but as a matter of fact, also when i have 20 agents training together in parallel, my CPU is at 30%, my GPU is at 3%.

    any suggestions on how to use as much as possible the GPU??

    thanks
     
  2. MrWetsnow

    MrWetsnow

    Joined:
    Jan 5, 2020
    Posts:
    60
  3. kahabal

    kahabal

    Joined:
    Nov 19, 2018
    Posts:
    16
    thanks