專案場景：（pytorch）CUDA ERROR:out of memory（已解決）
提示：這里簡述專案相關背景：
例如：專案場景：示例:通過藍牙芯片(HC-05)與手機 APP 通信，每隔 5s 傳輸一批傳感器資料(不是很大)
問題描述：新手調程式時，經常出現這種cuda錯誤無法解決，這里我總結一些了解決方法
提示：這里描述專案中遇到的問題：
例如：資料傳輸程序中資料不時出現丟失的情況，偶爾會丟失一部分資料
APP 中接收資料代碼：
@Override public void run() { bytes = mmInStream.read(buffer); mHandler.obtainMessage(READ_DATA, bytes, -1, buffer).sendToTarget(); } 解決方案：
1，換一個gpu，或者增加一個gpu，如：CUDA_VISIBLE_DEVICES=0，1（注意在改變gpu的時候，代碼里device設備號也需要修改）
2，減小num_worker和batch-size數目
3，pin_memory=True（這個問題最難發現，在此記錄一下），查看一下你們程式的pin_memory是不是True
轉載請註明出處，本文鏈接：https://www.uj5u.com/qianduan/187541.html
標籤：其他
上一篇：四大CPU架構的區別