项目场景：（pytorch）CUDA ERROR:out of memory（已解决）

 

---

问题描述：新手调程序时，经常出现这种cuda错误无法解决，这里我总结一些了解决方法

 

---

解决方案：

 

1，换一个gpu，或者增加一个gpu，如：CUDA_VISIBLE_DEVICES=0，1（注意在改变gpu的时候，代码里device设备号也需要修改）

2，减小num_worker和batch-size数目

3，pin_memory=True（这个问题最难发现，在此记录一下），查看一下你们程序的pin_memory是不是True