1 RuntimeError: CUDA error: no kernel image is available for execution on the device。
报错信息:安装带gpu的pytorch的时候,
(1)先警告
NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75.
If you want to use the NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
(2)紧接着报错执行深度学习代码报错:
RuntimeError: CUDA error: no kernel image is available for execution on the device。
原因: 是因为pytorch1.6的版本不支持gpu硬件RTX 3090,需要下载更高版本的pytorch才行。
解决: 下载1.8以上版本的pytorch,算力要支持
// 查看包支持的算力
import torch<

文章讲述了在使用PyTorch时遇到CUDAerror,特别是nokernelimageisavailable的问题,原因在于PyTorch版本1.6不支持NVIDIAGeForceRTX3090。解决方法是升级到1.8及以上版本以匹配高算力硬件。
最低0.47元/天 解锁文章
2619

被折叠的 条评论
为什么被折叠?



