JennieGerhardt 2019-11-28 10:28 采纳率: 0%
浏览 1121

cupy.cuda.memory.OutOfMemoryError

请教:
训练lda2vec的时候

gpu_id = int(os.getenv('CUDA_GPU', '0'))
cuda.get_device(gpu_id).use()
print("Using GPU " + str(gpu_id))

图片说明

训练几十个epoch以后cupy.cuda.memory.OutOfMemoryError,然后又重头开始训练

data = prepare_topics(cuda.to_cpu(model.mixture.weights.W.data).copy(),
                          cuda.to_cpu(model.mixture.factors.W.data).copy(),
                          cuda.to_cpu(model.sampler.W.data).copy(),
                          words)
  • 写回答

1条回答 默认 最新

  • threenewbee 2019-11-28 10:51
    关注

    内存不够,你的内存是不是遇到了2GB上限?这可能是因为你的pycharm或者python解释器是32bit的,你可以换notebook和64bit python
    如果还遇到问题,就是物理内存不足。

    评论

报告相同问题?

悬赏问题

  • ¥15 求差集那个函数有问题,有无佬可以解决
  • ¥15 【提问】基于Invest的水源涵养
  • ¥20 微信网友居然可以通过vx号找到我绑的手机号
  • ¥15 寻一个支付宝扫码远程授权登录的软件助手app
  • ¥15 解riccati方程组
  • ¥15 display:none;样式在嵌套结构中的已设置了display样式的元素上不起作用?
  • ¥15 使用rabbitMQ 消息队列作为url源进行多线程爬取时,总有几个url没有处理的问题。
  • ¥15 Ubuntu在安装序列比对软件STAR时出现报错如何解决
  • ¥50 树莓派安卓APK系统签名
  • ¥65 汇编语言除法溢出问题