解决error:CUDA error: no kernel image is available for execution on the device

1. 问题概述

在python中利用pytorch+cuda调用GPU进行训练的时候,报错:


error

2. 社区资料

该报错比较常见,已有多个社区博客提供了解决方案:

2.1 by 程序员行者孙

【深度学习故障排除】解决 CUDA “no kernel image is available“ 错误的指南_no kernel image is available for execution on the -CSDN博客
该博客提示error原因是因为pytorch,cuda与当前GPU不兼容,可以通过1. 查询GPU型号及其算力:CUDA GPUs - Compute Capability | NVIDIA Developer,然后根据算力查询对应cuda版本:1. Introduction — NVIDIA Data Center Drivers r570 documentation,然后安装对应版本cuda:Previous PyTorch Versions | PyTorch

2.2 by JiehoonKwak

No GPU/TPU found in RTX 4060 · Issue #19 · BayraktarLab/cell2fate
cell2fate工具默认使用1.11版本的pytorch以及10.2版本的cuda,并不兼容40系的显卡,因此需要对pytorch和cuda进行升级,步骤不赘述。

3. 总结

常见的报错。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容