darts
darts copied to clipboard
RuntimeError: cannot pin 'torch.cuda.DoubleTensor' on GPU on version 0.10.0
I'm trying to fit the RNN model on GPU and this is the error that I get:
RuntimeError: cannot pin 'torch.cuda.DoubleTensor' only dense CPU tensors can be pinned
I also ran the example from here and the same error. The fit
method runs without problem on the CPU but not on GPU.
Here's the error's backtrace: https://paste.ubuntu.com/p/KkpthPvPKp/
It seems that this problem has been introduced in the latest version (0.10.0), and it is not an issue in version 0.9.1 in which my code was written previously.