碼上快樂

相關內容    簡體    繁體

Pytorch: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMa

本文轉載自   查看原文   2019-04-04 10:49   10561    深度學習/ 雜七雜八


更換了數據集, 在計算交叉熵損失時出現錯誤 :
cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMa

解決檢查兩個問題:

1. 模型輸出label數量 是否與 標簽類別數量相同

<label為 男,女, 未知;  但是我模型設計只有男/女, 也就是label的標記數字超出了模型分類數>

2. label是否有-1的情況,需要提前過濾掉,類似下面這樣的標簽


免責聲明！

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。



猜您在找 Pytorch報錯：cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26 RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26 RuntimeError: cuda runtime error (59) : device-side assert triggered（已解決） CUDA error: device-side assert triggered RuntimeError: CUDA error: device-side assert triggered的解決 完美解決-RuntimeError: CUDA error: device-side assert triggered 解決RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:70&NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the late RuntimeError: size mismatch, m1: [32 x 2304], m2: [400 x 100] at C:/w/1/s/windows/pytorch/aten/src\THC/generic/THCTensorMathBlas.cu:268 Pytorch的runtime error pytorch 使用gpu報錯CUDA error: no kernel image is available for execution on the device
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM