• 工资「喂饱肚子」,副业「养活灵魂」!

  • 在本站开通年度VIP,无限制下载本站资源和阅读本站文章

  • 如果觉得本站的内容有帮助,可以考虑打赏博主哦!

  • 本站由于前段时间遭受到大量临时和国外邮箱注册,所以对可注册的邮箱类型进行了限制!

  • 本站会放置Google广告用于维持域名以及网站服务器费用。

  • 问题反馈可发送邮件到stubbornhuang@qq.com

  • 感谢大家访问本站,希望本站的内容可以帮助到大家!

  • 计算机图形学与计算几何经典必备书单整理,下载链接可参考:https://www.stubbornhuang.com/1256/

  • 欢迎大家交换友链,可在https://www.stubbornhuang.com/申请友情链接进行友链交换申请!

Python – 修复cupy.cuda.compile_with_cache deprecated错误

Python 发布于2025-02-27 阅读 1,130次 0次评论 0次点赞 本文共532个字,阅读需要2分钟。

1 修复cupy.cuda.compile_with_cache deprecated错误

最近在跑https://github.com/alex04072000/FuSta,以下的代码

@cupy.memoize(for_each_device=True)
def cupy_launch(strFunction, strKernel):
    return cupy.cuda.compile_with_cache(strKernel).get_function(strFunction)

出现了cupy.cuda.compile_with_cache deprecated的问题,这是由于在项目中安装了cupy的高版本,导致不兼容原有的函数。

上述代码可以进行如下修改:

@cupy.memoize(for_each_device=True)
def cupy_launch(strFunction, strKernel):
    module = cupy.RawModule(code=strKernel)
    return module.get_function(strFunction)

即可解决。

参考链接

欢迎扫码关注我的微信公众号,及时获取文章更新

微信公众号二维码

本文作者:StubbornHuang

版权声明:本文为站长原创文章,如果转载请注明原文链接!

原文标题:Python – 修复cupy.cuda.compile_with_cache deprecated错误

原文链接:https://www.stubbornhuang.com/3125/

发布于:2025年02月27日 15:59:44

修改于:2025年02月27日 16:00:30

声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。

文章末尾
上一篇
python - 减少opencv-python保存的视频文件大小
Python
下一篇
视频智能分镜/视频转场检测模型TransNetV2导出为onnx和基于onnxruntime的python推理类
视频智能分镜
当前分类随机文章推荐

发表评论

您必须 [ 登录 ] 才能发表留言!

关注我们的公众号

微信公众号