euler icon indicating copy to clipboard operation
euler copied to clipboard
verified publisher icon alibaba

GPU 上跑显存爆掉，有没有办法embedding在cpu上，计算在GPU上呀

Open pangedeshijie opened this issue 6 years ago • 1 comments

我GPU的显存不够，有没有办法，整个embedding在cpu上，每个batch对应的embedding送到GPU上计算？

pangedeshijie avatar Dec 27 '19 02:12 pangedeshijie

我也遇到了这个问题，不知道单机情况下如何实现这个？

David-AJ avatar Aug 13 '20 06:08 David-AJ