1. 程式人生 > 其它 >pytorch實踐-bug/報錯記錄

pytorch實踐-bug/報錯記錄

1. RuntimeError: CUDA error: device-side assert triggered

原因: 通常是由陣列越界造成!仔細檢查就能發現

注意: 一般情況,程式報錯的程式碼和真正的出錯程式碼不是同一行
解決辦法:

程式碼加上:

import os
os.environ['CUDA_LAUNCH_BLOCKING'] = "0"

轉自 https://blog.csdn.net/Hungryof/article/details/97248209