Skip to content

[DOCS] Reload the library that could retain global gpu resources#8760

Closed
tqchen wants to merge 1 commit intoapache:mainfrom
tqchen:docs
Closed

[DOCS] Reload the library that could retain global gpu resources#8760
tqchen wants to merge 1 commit intoapache:mainfrom
tqchen:docs

Conversation

@tqchen
Copy link
Member

@tqchen tqchen commented Aug 16, 2021

@tqchen
Copy link
Member Author

tqchen commented Aug 16, 2021

background sphinx-gallery/sphinx-gallery#853

@mbs-octoml
Copy link
Contributor

I didn't have any luck using just the module name nor explicitly removing all tensorflow* modules as they do for 'seaborn' (the latter triggers an internal tensorflow error). I've also tried launching the tensorflow tutorial as a sub-process but that didn't seem to actually separate out the module loads.

"min_reported_time": 60,
"expected_failing_examples": [],
"reset_modules": (force_gc, "matplotlib", "seaborn"),
"reset_modules": (force_gc, "matplotlib", "tensorflow", "torch", "onnx"),
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@tqchen
Copy link
Member Author

tqchen commented Aug 16, 2021

Good catch @areusch Seems indeed this would need some special support from sphinx gallery side.

Perhaps we could reuse the same strategy here
https://github.com/sphinx-gallery/sphinx-gallery/blob/b41e328230f016b2089464b8f834fd76a6395eac/sphinx_gallery/scrapers.py#L550

@tqchen tqchen closed this Aug 16, 2021
@tqchen
Copy link
Member Author

tqchen commented Aug 16, 2021

closing for now as it is more complicated than I initially thought

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants