You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
@ybzhou hi, I've only tested it on single GPU.
the issue might be here, where the CUDA steam is created once and fixed when the file is imported. this means it's binded to a single GPU probably.
need to investigate how to change the stream. let me know if you have a fix :)
Now there is a version in "multigpu" branch. I will do more test before merging it into the master branch. Please feel free to use the current version.
When using with DataParallel, received the following error
cupy.cuda.driver.CUDADriverError: CUDA_ERROR_INVALID_HANDLE: invalid resource handle
single gpu works fine
The text was updated successfully, but these errors were encountered: