Skip to content

Instantly share code, notes, and snippets.

@rnyak
Last active July 22, 2020 20:22
Show Gist options
  • Save rnyak/422ec6c4a7b554c18781251ab3fb2311 to your computer and use it in GitHub Desktop.
Save rnyak/422ec6c4a7b554c18781251ab3fb2311 to your computer and use it in GitHub Desktop.
rmm.reinitialize(pool_allocator=True, initial_pool_size=0.8 * rmm.get_info().free)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment