使用pytorch时安装cuda

一叶清风扬

于 2023-05-16 21:06:28 发布

阅读量3.2k

点赞数

文章标签： pytorch python cuda

本文链接：https://blog.csdn.net/kevinshift/article/details/130714132

版权

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

基本情况

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,

最低0.47元/天解锁文章