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)