You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/08/12 18:43:41 UTC

[GitHub] [tvm] tqchen commented on pull request #8722: Force a gc between sphinx-gallery items to reclaim GPU memory.

tqchen commented on pull request #8722:
URL: https://github.com/apache/tvm/pull/8722#issuecomment-897881477


   Some followup context
   
   appears that sphinx-gallery did recycle global variables, i also did a quick exp to confirm that
   
   so the memory contention might due to the global retention of packages that we use(likely Tensorflow or pytorch who might retain its internal memory pool)
   
   Adding "tensorflow", "torch" to reset_modules might be the right solution 


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org