报错
RuntimeError: CUDA error: no kernel image is available for execution on the device
一般是显卡与cuda或者pytorch不适配导致的,可以查看本机或者服务器的cuda版本
nvidia-smi![]()
再到pytorch官网:PyTorch 安装对应版本即可
版权声明:本文为qq_38964360原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
报错
RuntimeError: CUDA error: no kernel image is available for execution on the device
一般是显卡与cuda或者pytorch不适配导致的,可以查看本机或者服务器的cuda版本
nvidia-smi![]()
再到pytorch官网:PyTorch 安装对应版本即可