2020-10-21

项目场景：（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

来源：oschina

链接：https://my.oschina.net/u/4414278/blog/4686067