如题,我想测试一组1亿以上的数据,得到随机数值,然后经过计算求得平均值,但是当数据达到1亿时就会报out of memory的错误,该怎么解决呢?要是用循环,循环怎么用呢?
你可以依据你的显存的大小,每次生成N个数据(N能够放入你的显存中),然后计算平均值;计算完成后再生成另外N个,接着计算,以此类推。
通俗和通用的讲,将你的任务切分成多个小任务,这样每个小任务都能在显存里放下。
嗯嗯,我已经解决啦,谢谢风老大~
哈,千军兄,我已解决啦,谢谢啦
如题,我想测试一组1亿以上的数据,得到随机数值,然后经过计算求得平均值,但是当数据达到1亿时就会报out of memory的错误,该怎么解决呢?要是用循环,循环怎么用呢?
你可以依据你的显存的大小,每次生成N个数据(N能够放入你的显存中),然后计算平均值;计算完成后再生成另外N个,接着计算,以此类推。
通俗和通用的讲,将你的任务切分成多个小任务,这样每个小任务都能在显存里放下。
嗯嗯,我已经解决啦,谢谢风老大~
哈,千军兄,我已解决啦,谢谢啦