RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at torch/csrc/cuda/Module.cpp:51

Question

当我尝试加载 pytorch 检查点时：

checkpoint = torch.load(pathname)

我明白了：

RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at 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)