dmlc / dmlc/MXNet.jl

Julia garbage collector does not reclaim GPU memory

Open
#85 9 comments 0 reactions 0 assignees View on GitHub
Dominant language
No language data
Stars
369
Forks
63
PR merge metrics
No merged PRs in 30d

Description

I'm trying to write Neural Style in MXNet.jl, and I keep running out of memory when I try to make new executors (and delete the old ones). My basic strategy is to store the executor in an `exec` variable and do

`exec = 0`
`gc()`

when I want to reclaim GPU memory for that executor. This does not work as expected, as I am tracking CUDA memory usage with `nvidia-smi` and there is never a drop in memory usage after calling `gc()`.

Does anyone know of a way to reclaim GPU memory? Here is my code for reference: https://github.com/bonsairobo/mxnet-neural-style/blob/master/stylenet.jl

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.