Блог

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

 checkpoint = torch.load(pathname)
 

Я вижу:

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

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

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

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

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

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