如何识别chainer中的GPU内存瓶颈

Question

我有一个chainer模型，有时因cupy.cuda.memory.OutOfMemoryError而崩溃。由于它发生的确切位置取决于批次元素的大小，我想知道是否有办法识别chainer模型中的内存瓶颈？

Answer 1

您可以参考/home/jturolla/Documents/btcbot/manager.js:73 console.log(output.adx); ^ TypeError: Cannot read property 'adx' of undefined at updateIndicators (/home/jturolla/Documents/btcbot/manager.js:73:22) at Manager.runBot (/home/jturolla/Documents/btcbot/manager.js:54:7) at Object.<anonymous> (/home/jturolla/Documents/btcbot/app.js:4:9) at Module._compile (internal/modules/cjs/loader.js:702:30) at Object.Module._extensions..js (internal/modules/cjs/loader.js:713:10) at Module.load (internal/modules/cjs/loader.js:612:32) at tryModuleLoad (internal/modules/cjs/loader.js:551:12) at Function.Module._load (internal/modules/cjs/loader.js:543:3) at Function.Module.runMain (internal/modules/cjs/loader.js:744:10) at startup (internal/bootstrap/node.js:238:19) at bootstrapNodeJSCore (internal/bootstrap/node.js:572:3) 。

• https://github.com/chainer/chainer/blob/master/chainer/function_hooks/cupy_memory_profile.py#L17

CupyMemoryProfileHook