自己写的程序遇到了如下错误:
“RuntimeError: cuda runtime error (2) : out of memory at d:\downloads\pytorch-master-1\torch\lib\thc\generic/THCStorage.cu:66”
cuda有一定的运行时间限制,超过一定的时间就认为超时报错。
这里提供一些解决方案:

减少load数据的时间。
可能load的时间也被算进去了,我的数据很大的时候Memory显示100%,卡住了.
于是提前写一个数据处理脚本,提前将数据写入csv,要用的时候直接拿取就好了。
减少数据/减少计算量
可以推算一下网络需要的参数个数,每个占4Bytes.
需要存储的参数个数*4Byte/(1024^3)
算出来就是你需要多少G的存储空间了。
查看一下你的GPU可以容纳多少空间,看看是不是真的超了。超了的话只好减少数据/简化网络结构从而减少计算量了。
很棒呀
很棒呀.继续加油