会员
周边
捐助
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Dechin的博客
研究兴趣: 分子动力学模拟、量子计算与量子算法、Python编程与深度学习
博客园
首页
新随笔
联系
订阅
管理
2024年4月11日
MindSpore编译构建后Vmap模块的RuntimeError问题
摘要:
在MindSpore编译计算图的过程中,会把从编译构建好的whl包中引入的模块视为第三方库,也就没有办法在即时编译的阶段入图。普通的math和numpy等第三方库不入图也不会影响计算。但如果是基于MindSpore本身开发的一些函数,如果用到了Jit、Grad和Vmap,那么有可能出现无法入图的问题,就会出现RuntimeError报错。解决方法就是设置一个跟即时编译有关的环境变量,把相关的第三方包引用修改为内部引用。
阅读全文
posted @ 2024-04-11 10:24 DECHIN
阅读(142)
评论(0)
推荐(0)
编辑
公告