PyTorch报CUDA error: no kernel image is available for execution on the device问题解决

最新推荐文章于 2024-06-07 10:43:26 发布

奥特曼熬夜不sleep

最新推荐文章于 2024-06-07 10:43:26 发布

阅读量2.3k

点赞数 2

文章标签： pytorch

本文链接：https://blog.csdn.net/m0_45275062/article/details/119361464

版权

首先讲一下报错的原因PyTorch高版本对显卡算力是有所要求的，驱动程序只是从软件角度去利用显卡，显卡的算力是不会改变的，报错就在于显卡的算力太低。这里就涉及到了NVIDIA GPU CUDA的compute capability，这是一个类似版本的概念。不同型号的GPU的compute capability不同。
其次解决办法就是使用低版本的PyTorch,或者更换新显卡。目前官方列出的清单（GPU Compute Capability）上，较低的为2.1，都是较早的GPU型号，如：GT430, GT630, GT730 DDR3 128bit；较高的可达7.5，都是较新的GPU型号，如：RTX 2080, RTX 2070等。下面给出显卡算力的查看方法：

NVIDIA开发者网站上的CUDA GPU Compute Capability列表：

GPU Compute Capability

关注