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

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

checkpoint = torch.load(pathname)

Я понимаю:

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

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

Как мне загрузить КПП?


python pytorch checkpointing
person Tom Hale    schedule 19.04.2019    source источник

Ответы (1)


arrow_upward
7
arrow_downward

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

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

checkpoint = torch.load(pathname, map_location=map_location)
person Tom Hale    schedule 19.04.2019