deep-rl-class icon indicating copy to clipboard operation
deep-rl-class copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

[HANDS-ON BUG] mlagents-learn in unit 5 not working

Open benbekir opened this issue 1 year ago • 4 comments

Describe the bug

The command

!mlagents-learn ./config/ppo/SnowballTarget.yaml --env=./training-envs-executables/linux/SnowballTarget/SnowballTarget --run-id="SnowballTarget1" --no-graphics

doesnt work. It always results in the following error:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument mat1 in method wrapper_CUDA_addmm).

I have tried specifying the device with the --torch-device argument but that didnt help either. Maybe this has something to do with the fact that dev versions are used for the ml-agents and ml-agents-envs packages?

  ml-agents: 1.2.0.dev0,
  ml-agents-envs: 1.2.0.dev0,
  Communicator API: 1.5.0,
  PyTorch: 2.5.0+cu121

Material

  • Did you use Google Colab? Yes

benbekir avatar Oct 24 '24 11:10 benbekir