Open benbekir opened 2 days ago
Experiencing the same issue. Additionally, the runtime requires a restart after the installation of numpy packages, otherwise it can't find the hyperparameter file in /config/ppo/SnowballTarget.yaml. Maybe it's a package version conflict?
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?Material