[Bug Report] torch_to_numpy
fails when tensors are not on cpu
#1108
Labels
bug
Something isn't working
torch_to_numpy
fails when tensors are not on cpu
#1108
Describe the bug
torch_to_numpy fails when
torch.Tensor
is on a non-cpu device or hasrequires_grad
set toTrue
.Note
Calling torch_to_jax does not cause these problems when
jax
is installed with GPU support - otherwise it also throws an error but informs the user thatjax
should be installed with GPU support.Code example
System info
python=3.11
,gymnasium=1.0.0a2
Additional context
Error traceback from the example script:
Checklist
The text was updated successfully, but these errors were encountered: