[debug] RuntimeError: CUDA error: no kernel image is available for execution on the device

问题描述

运行程序时出现报错：

RuntimeError: CUDA error: no kernel image is available for execution on the device (launch_kernel at /opt/conda/conda-bld/pytorch_1556653114079/work/aten/src/ATen/native/cuda/Loops.cuh:72)

问题原因

当前使用的硬件算力不支持相应的pytorch函数，需要更换pytorch到一个合适的版本。

解决方法

查看当前设备的算力
在官网查看即可

查看当前pytorch版本支持的算力

 torch.cuda.get_arch_list()

Python 3.7.10 (default, Jun  4 2021, 14:48:32)
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.get_arch_list()
['sm_37', 'sm_50', 'sm_60', 'sm_70', 'sm_75']

选择合适的pytorch版本，支持当前的GPU算力

附录：完整报错信息

Traceback (most recent call last):File "work/main.py", line 171, in <module>alg(args.input_dir, args.output_dir,args)File "work/main.py", line 127, in algoutputs = single_gpu_test(model, data_loader,save_path =split_result )File "/workspace/work/tools/TianZhi_test.py", line 29, in single_gpu_testresult = model(return_loss=False, rescale=not show, **data)File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__result = self.forward(*input, **kwargs)File "/opt/conda/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 150, in forwardreturn self.module(*inputs[0], **kwargs[0])File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__result = self.forward(*input, **kwargs)File "/workspace/work/mmdet/core/fp16/decorators.py", line 49, in new_funcreturn old_func(*args, **kwargs)File "/workspace/work/mmdet/models/detectors/base.py", line 88, in forwardreturn self.forward_test(img, img_meta, **kwargs)File "/workspace/work/mmdet/models/detectors/base.py", line 79, in forward_testreturn self.simple_test(imgs[0], img_metas[0], **kwargs)File "/workspace/work/mmdet/models/detectors/polarmask.py", line 79, in simple_testbbox_list = self.bbox_head.get_bboxes(*bbox_inputs)File "/workspace/work/mmdet/core/fp16/decorators.py", line 127, in new_funcreturn old_func(*args, **kwargs)File "/workspace/work/mmdet/models/anchor_heads/polarmask_last_head.py", line 603, in get_bboxesscale_factor, cfg, rescale)File "/workspace/work/mmdet/models/anchor_heads/polarmask_last_head.py", line 691, in get_bboxes_singlecfg.max_per_img,)File "/workspace/work/mmdet/core/post_processing/bbox_nms.py", line 111, in multiclass_nms_with_maskcls_dets, index = nms_op(cls_dets, **nms_cfg_)File "/workspace/work/mmdet/ops/nms/nms_wrapper.py", line 43, in nmsinds = nms_cuda.nms(dets_th, iou_thr)
RuntimeError: CUDA error: no kernel image is available for execution on the device (launch_kernel at /opt/conda/conda-bld/pytorch_1556653114079/work/aten/src/ATen/native/cuda/Loops.cuh:72)
frame #0: c10::Error::Error(c10::SourceLocation, std::string const&) + 0x45 (0x7f38c8b12dc5 in /opt/conda/lib/python3.7/site-packages/torch/lib/libc10.so)
frame #1: void at::native::gpu_index_kernel<__nv_dl_wrapper_t<__nv_dl_tag<void (*)(at::TensorIterator&, c10::ArrayRef<long>, c10::ArrayRef<long>), &(void at::native::index_kernel_impl<at::native::OpaqueType<8> >(at::TensorIterator&, c10::ArrayRef<long>, c10::ArrayRef<long>)), 1u>> >(at::TensorIterator&, c10::ArrayRef<long>, c10::ArrayRef<long>, __nv_dl_wrapper_t<__nv_dl_tag<void (*)(at::TensorIterator&, c10::ArrayRef<long>, c10::ArrayRef<long>), &(void at::native::index_kernel_impl<at::native::OpaqueType<8> >(at::TensorIterator&, c10::ArrayRef<long>, c10::ArrayRef<long>)), 1u>> const&) + 0x33e (0x7f38ce75013e in /opt/conda/lib/python3.7/site-packages/torch/lib/libcaffe2_gpu.so)
frame #2: <unknown function> + 0x27e9cca (0x7f38ce74bcca in /opt/conda/lib/python3.7/site-packages/torch/lib/libcaffe2_gpu.so)
frame #3: <unknown function> + 0x27ea555 (0x7f38ce74c555 in /opt/conda/lib/python3.7/site-packages/torch/lib/libcaffe2_gpu.so)
frame #4: <unknown function> + 0x6cb2aa (0x7f38c93f72aa in /opt/conda/lib/python3.7/site-packages/torch/lib/libcaffe2.so)
frame #5: at::native::index(at::Tensor const&, c10::ArrayRef<at::Tensor>) + 0x3e8 (0x7f38c93f4ed8 in /opt/conda/lib/python3.7/site-packages/torch/lib/libcaffe2.so)
frame #6: at::TypeDefault::index(at::Tensor const&, c10::ArrayRef<at::Tensor>) const + 0x6c (0x7f38c97beb0c in /opt/conda/lib/python3.7/site-packages/torch/lib/libcaffe2.so)
frame #7: torch::autograd::VariableType::index(at::Tensor const&, c10::ArrayRef<at::Tensor>) const + 0x6f9 (0x7f38c5ec5289 in /opt/conda/lib/python3.7/site-packages/torch/lib/libtorch.so.1)
frame #8: at::Tensor::index(c10::ArrayRef<at::Tensor>) const + 0x59 (0x7f3849a5599d in /workspace/work/mmdet/ops/nms/nms_cuda.cpython-37m-x86_64-linux-gnu.so)
frame #9: nms_cuda(at::Tensor, float) + 0x779 (0x7f3849a5414d in /workspace/work/mmdet/ops/nms/nms_cuda.cpython-37m-x86_64-linux-gnu.so)
frame #10: nms(at::Tensor const&, float) + 0x130 (0x7f3849a46e10 in /workspace/work/mmdet/ops/nms/nms_cuda.cpython-37m-x86_64-linux-gnu.so)
frame #11: <unknown function> + 0x2084d (0x7f3849a5284d in /workspace/work/mmdet/ops/nms/nms_cuda.cpython-37m-x86_64-linux-gnu.so)
frame #12: <unknown function> + 0x1df0a (0x7f3849a4ff0a in /workspace/work/mmdet/ops/nms/nms_cuda.cpython-37m-x86_64-linux-gnu.so)
<omitting python frames>

[debug] RuntimeError: CUDA error: no kernel image is available for execution on the device相关推荐

RTX 3090运行报错：RuntimeError: CUDA error: no kernel image is available for execution on the device
RuntimeError: CUDA error: no kernel image is available for execution on the device 安装适用于GeForce RTX ...
RuntimeError: CUDA error: no kernel image is available for execution on the device
导致的原因一般都是显卡算力和cuda或者torch版本不匹配比如在conda中安装的pytorch=1.5.0 cuda=10.2 错误:RuntimeError: CUDA error: no k ...
出现错误（已解决）RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA ker
出现问题(已解决)RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA ker ...
（ubuntu）YOLOv5报错：RuntimeError: CUDA error: no kernel image is available for execution on the ...
笔者之前在另一台电脑成功跑通了自己修改的yolov5,但在新系统运行时报了如下错误 RuntimeError: CUDA error: no kernel image is available for ...
显卡RTX 3090运行pytorch报错CUDA error: no kernel image is available for execution on the device
RuntimeError: CUDA error: no kernel image is available for execution on the device 安装适用于GeForce RTX ...
解决CUDA error: no kernel image is available for execution on the device
1. 电脑配置 GPU 3080 算力8.6 CUDA 11.1 CUDNN 8.2.0 conda 4.9.2 python 3.8.5 2. 问题描述首先在pytroch官网根据电脑相应配置使用 ...
RuntimeError: CUDA error: no kernel image is available for execution on the driver
问题:RuntimeError: CUDA error: no kernel image is available for execution on the driver, when use pyto ...
PyTorch报CUDA error: no kernel image is available for execution on the device问题解决
首先讲一下报错的原因PyTorch高版本对显卡算力是有所要求的,驱动程序只是从软件角度去利用显卡,显卡的算力是不会改变的,报错就在于显卡的算力太低.这里就涉及到了NVIDIA GPU CUDA的com ...
GT820M显卡使用CUDA的“no kernel image is available for execution on the device”问题解决
最近使用比较老的电脑使用CUDA,一开始安装10.1的windows版本,发现无法使用,经过网上查阅信息发现是显卡驱动不支持,电脑的NVIDIA驱动是390. 然后查询到该驱动支持9.0版本的CUDA ...

[debug] RuntimeError: CUDA error: no kernel image is available for execution on the device

问题描述

问题原因

解决方法

附录：完整报错信息

[debug] RuntimeError: CUDA error: no kernel image is available for execution on the device相关推荐

最新文章

热门文章