使用pytorch时安装cuda

最新推荐文章于 2025-10-11 10:53:15 发布

原创

最新推荐文章于 2025-10-11 10:53:15 发布 · 3.3k 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #cuda

当尝试在没有CUDA的环境中加载使用GPU保存的PyTorch模型时，会遇到错误。解决方法是在加载模型时指定`map_location=cpu`，将模型映射到CPU上。另外，可以使用特定CUDA版本的pip命令安装PyTorch，例如针对CUDA11.3的版本。测试CUDA支持性，可以使用`torch.cuda.is_available()`函数。

基本情况

pytorch有基于cpu和gpu运行的，gpu时需要cuda。所以需要再装cuda。

如果没安装cuda，则运行下面的模型加载时会报错：

model = torch.load(model_path)

报错：

  File "test.py", line 43, in __init__
    model = torch.load(model_path)
  File "C:\Users\86137\Anaconda3\envs\Pytorch\lib\site-packages\torch\serialization.py", line 607, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "C:\Users\86137\Anaconda3\envs\Pytorch\lib\site-packages\torch\serialization.py", line 882,