Ошибка выполнения: ошибка времени выполнения cuda (35): версии драйвера CUDA недостаточно для версии времени выполнения CUDA в torch / csrc / cuda / Module.cpp: 51

Question

Когда я пытаюсь загрузить контрольную точку pytorch:

checkpoint = torch.load(pathname)

Я вижу:

RuntimeError: ошибка времени выполнения cuda (35): версии драйвера CUDA недостаточно для времени выполнения CUDAверсия в torch / csrc / cuda / Module.cpp: 51

Я создал контрольную точку с доступным графическим процессором, но теперь доступен только центральный процессор.

Как загрузить контрольную точку

Answer 1

Загрузить данные контрольной точки в лучшее доступное на данный момент местоположение:

if torch.cuda.is_available():
    map_location=lambda storage, loc: storage.cuda()
else:
    map_location='cpu'

checkpoint = torch.load(pathname, map_location=map_location)