AssertionError: Torch not compiled with CUDA enabled 解决方案

报错

Traceback (most recent call last):File "main_pretrain.py", line 250, in <module>main(args)File "main_pretrain.py", line 185, in mainmodel.to(device)File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 899, in toreturn self._apply(convert)File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 570, in _applymodule._apply(fn)File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 570, in _applymodule._apply(fn)File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 593, in _applyparam_applied = fn(param)File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\nn\modules\module.py", line 897, in convertreturn t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)File "D:\APP\Anaconda3\envs\py37\lib\site-packages\torch\cuda\__init__.py", line 208, in _lazy_initraise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

解决

安装cuda

查看本地CUDA支持的版本

nvidia-smi

如红框所示，电脑支持的最高CUDA版本为11.4

检查自己的电脑中CUDA的版本（检查是否安装CUDA）：

打开cmd，输入

nvcc --version

理论上应显示当前安装的CUDA版本
若显示nvcc不存在，则说明未安装CUDA

进入官网下载CUDA
建议下载11.3版本
官网地址：https://developer.nvidia.com/cuda-downloads
下载历史版本：https://developer.nvidia.com/cuda-toolkit-archive
找到对应版本进行下载
选择对应系统
安装后使用nvcc--version即可看到cuda版本
下载 cuDNN

地址：https://developer.nvidia.com/rdp/cudnn-download

下载cuDNN是需要登录英伟达开发者账户

注意：必须选择和安装的CUDA匹配的版本

解压

复制bin下的文件 cudnn64_8.dll 到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin

添加环境变量
运行test

AssertionError: Torch not compiled with CUDA enabled 解决方案相关推荐

AssertionError: Torch not compiled with CUDA enabled解決方案
在執行pytorch代碼的時候,突然看到報錯 AssertionError: Torch not compiled with CUDA enabled 這說明了 1. 你pytoch确实安装了 2. ...
成功解决raise AssertionError(“Torch not compiled with CUDA enabled“)AssertionError: Torch not compiled
成功解决raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not ...
AssertionError: Torch not compiled with CUDA enabled到底是什么原因
代码运行报错AssertionError: Torch not compiled with CUDA enabled 但是根据cuda安装显示 cuda是安装成功的这是怎么回事我先进入百度翻译 A ...
解决 AssertionError Torch not compiled with CUDA enabled
最近在矩池云的的Tesla K80机子上跑MMYOLO,跟着MMYOLO官方文档<自定义数据集标注+训练+测试+部署全流程 >操作到 "2.1.1 软件或者算法辅助" ...
报错 AssertionError: Torch not compiled with CUDA enabled
报错: AssertionError: Torch not compiled with CUDA enabled 本文针对仅有CPU无GPU的设备,针对该报错进行修改: 在代码开头添加: // A c ...
AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】
目录问题查阅资料解决步骤 1.检查pytorch版本.是否有CUDA 2.安装CUDA前看电脑的显卡驱动程序版本.支持的最高版本 3.安装CUDA和cuDNN 4.卸载pytorch 5.重新安 ...
问题：AssertionError: Torch not compiled with CUDA enabled
问题在于下载的torch没有cuda,即你的电脑没有GPU,无法运行解决办法:在程序开头添加下列代码,然后在出现.cuda()的地方换成.to(device) device = torch.devi ...
报错：AssertionError: Torch not compiled with CUDA enabled
环境:win0+torch(cpu) 详细报错信息: 解决: 查阅资料发现此错误是由于下载的torch没有cuda,所以在运行时会报错,在程序最开始的地方加上: device = torch.devi ...
成功解决：AssertionError: Torch not compiled with CUDA enabled
在运行pycharm项目的时候,出现了以上的报错,主要可以归结于以下两个个方面: 1.没有安装GPU版本的pytorch,只是使用清华的镜像地址下载了CPU版本的pytorch 2.安装的CUDA和安 ...

AssertionError: Torch not compiled with CUDA enabled 解决方案

报错

解决

安装cuda

AssertionError: Torch not compiled with CUDA enabled 解决方案相关推荐

最新文章

热门文章