ML-Agent not using full CPU/GPU

Unity Engine
,
1

Hello,
I am relatively new to ML-Agents and have a problem:
ML-Agents doesn’t use full CPU or GPU, when I run the command :
mlagents-learn config.yaml --run-id=Test1 --env=Build\Test_Build --num-envs=13 --torch-device=cpu
It only uses 50% of my CPU, when I change the param “–toch-device” to “cuda” It uses 15% of my GPU, but i think that’s because I only have a GT 1030 and it uses all the VRAM.
My config file is the default provided on the Github page.

Thanks for your help

2

@Ewu_Uwe did you find a solution?

1 Like