Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add tf gpu allocator env var to tox #1747

Merged
merged 2 commits into from
Jan 11, 2023

Conversation

jperez999
Copy link
Contributor

adds global env var to ensure all tensorflow imports only allocate the required memory and release after use.

@jperez999 jperez999 added this to the 23.01 milestone Jan 11, 2023
@github-actions
Copy link

Documentation preview

https://nvidia-merlin.github.io/NVTabular/review/pr-1747

@karlhigley karlhigley merged commit 196814a into NVIDIA-Merlin:main Jan 11, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants