原文: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