RuntimeError：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

我創建了具有可用GPU的檢查點，但現在只有CPU可用。

如何加載檢查點？

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)