Pytorch завершается сбоем из-за ошибки CUDA: в Colab срабатывает подтверждение на стороне устройства ⇐ Python

[Anonymous]

Я пытаюсь инициализировать тензор в Google Colab с включенным графическим процессором.

Код: Выделить всё

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

t = torch.tensor([1,2], device=device)

Но я получаю эту странную ошибку.

Код: Выделить всё

RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1

Даже если установить для этой переменной среды значение 1, никаких дополнительных подробностей не будет.

У кого-нибудь когда-нибудь была такая проблема?


Подробнее здесь: https://stackoverflow.com/questions/681 ... d-on-colab