栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Pytorch错误:Torch not compiled with CUDA enabled

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Pytorch错误:Torch not compiled with CUDA enabled

报错复现:
Traceback (most recent call last):
  File "../train.py", line 388, in 
    main()
  File "../train.py", line 384, in main
    train(args)
  File "../train.py", line 378, in train
    trainer.train()
  File "../train.py", line 322, in train
    self._init_model()
  File "../train.py", line 308, in _init_model
    self._set_device()
  File "../train.py", line 168, in _set_device
    self._model.to(self._device)
  File "D:AppsAnacondaenvstse_2021_baseslibsite-packagestorchnnmodulesmodule.py", line 425, in to
    return self._apply(convert)
  File "D:AppsAnacondaenvstse_2021_baseslibsite-packagestorchnnmodulesmodule.py", line 201, in _apply
    module._apply(fn)
  File "D:AppsAnacondaenvstse_2021_baseslibsite-packagestorchnnmodulesmodule.py", line 223, in _apply
    param_applied = fn(param)
  File "D:AppsAnacondaenvstse_2021_baseslibsite-packagestorchnnmodulesmodule.py", line 423, in convert
    return t.to(device, dtype if t.is_floating_point() else None, non_blocking)
  File "D:AppsAnacondaenvstse_2021_baseslibsite-packagestorchcuda__init__.py", line 196, in _lazy_init
    _check_driver()
  File "D:AppsAnacondaenvstse_2021_baseslibsite-packagestorchcuda__init__.py", line 94, in _check_driver
    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

可见在将model送到GPU的过程中报错,CUDA不可用。

解决:
import torch
print(torch.__version__)
print(torch.cuda.is_available())

通过以上代码查看已安装pytorch版本是否可以调用GPU
返回为False的话,去pytorch官网依照命令重新安装相应版本的pytorch
pytorch与cuda版本问题参考链接。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/300467.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号