Last active
January 23, 2022 03:19
-
-
Save dhruvaray/66c277777d8b47dea0d885c555c9be00 to your computer and use it in GitHub Desktop.
Use torch multiprocessing & share_memory APIs
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
https://pytorch.org/docs/stable/notes/multiprocessing.html#cuda-in-multiprocessing