原文:pytorch 使用gpu报错CUDA error: no kernel image is available for execution on the device

报错原因 cuda版本和 pytorch不匹配 解决办法 点击此连接 进入如下图所示的页面 查看cuda版本:nvidia smi 根据自己所需的torch 系统 cuda版本选择对应的安装方式即可 查看各版本对应关系 参考链接 ...

2021-06-17 11:35 0 865 推荐指数:

查看详情

pytorch查看CUDA支持情况,只需要三行代码,另附Cuda runtime error (48) : no kernel image is available for execution处理办法

上面的命令只是检测CUDA是否安装正确并能被Pytorch检测到,并没有说明是否能正常使用,要想看Pytorch能不能调用cuda加速,还需要简单的测试一下: 一般来讲,输出主要是报48号错误,也就是CUDA的问题,出现这个问题在于硬件的支持情况,对于算力3.0 ...

Thu Dec 20 23:14:00 CST 2018 3 10877
指定用第几块GPUCUDA error: invalid device ordinal、

在指定用第几块GPU的时候,在代码开头输入了 表示用第二块GPU,并且也只能看见第二块,然后输入: 这时候输出的就是1(其实一共有4块) 至此应该不会有什么问题,但是问题出就出在后面的代码: 最好用以下形式,而不是用.cuda(),因为这个无法指定 ...

Tue May 11 02:46:00 CST 2021 0 2989
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM