no kernel image is available

在使用到显卡跑pytorch相关代码时候,会出现这样的一个问题 no kernel image is available 。这个是两种情况导致的,在你使用setup build你的项目时候,pytorch自动检测你当前的gpu算力并将其作为目前你电脑gpu的算力,pytorch源码如下:   也就是说出现no kernel image is availabl 第一种情况你在a卡build,却在b卡
相关文章
相关标签/搜索