Я установил Cuda 10.1 и последнюю версию драйвера Nvidia для моего Geforce 2080 ti. Я пытаюсь запустить базовый скрипт, чтобы проверить, работает ли pytorch, и я получаю следующую ошибку:
RuntimeError: cuda runtime error (999) : unknown error at ..\aten\src\THC\THCGeneral.cpp:50
Ниже приведен код, который я пытаюсь запустить:
import torch
torch.cuda.current_device()
torch.cuda.is_available()
torch.cuda.get_device_name(0)