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. 总结
常见的报错。