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卡
相關文章
相關標籤/搜索