Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Crash after setting PyTorch default tensor type to cuda.FloatTensor

See original GitHub issue

Hello folks at Facebook, setting PyTorch default tensor type to cuda.FloatTensor leads Ax to crashing in a Colab GPU instance.

You can checkout a complete test setup in Colab. This is the culprit: torch.set_default_tensor_type(torch.cuda.FloatTensor)

Keep up the great work, Enrico Bonetti Vieno

Issue Analytics

State:
Created 4 years ago
Comments:5 (4 by maintainers)

Top GitHub Comments

2reactions

Balandatcommented, Jan 22, 2020

See https://github.com/pytorch/pytorch/issues/32494 for the issue and https://github.com/pytorch/pytorch/pull/32496 for the fix.

1reaction

ebonetticommented, Jan 30, 2020

Thank you for the fast resolution, great job! 💪

Top Results From Across the Web

Is there anything wrong with setting default tensor type to cuda?

I ran into a problem with allennlp 's ElmoEncoder when I had set the default tensor to cuda.FloatTensor . In case anyone else...

If I'm not specifying to use CPU/GPU, which one is my script ...

PyTorch defaults to the CPU, unless you use the .cuda() methods on your models and the torch.cuda.XTensor variants of PyTorch's tensors.

PyTorch vs Apache MXNet

To install Apache MXNet with GPU support, you need to specify CUDA version. ... Returns a copy of the tensor after casting to...

pytorch runtimeerror: expected scalar type long but found float - You ...

LongTensor is synonymous with integer. PyTorch won't accept a FloatTensor as categorical target, so it's telling you to cast your tensor to LongTensor...

PyTorch 1.6.0 Now Available | Exxact Blog

... first convert the tensor to a long tensor, then to float tensor. ... Fixed the crash problem when using BuildExtension.with_options ...