Open Haoxiang-Wang opened 1 year ago
I've found a solution for purposes, and hopefully it works for anyone having this problem.
Looks like the device is hardcoded to be "cuda:0" in various parts of the code, leading to conflict. I set device="cuda:1" in optvis.transform, optvis.param.spatial, and optvis.param.color, and everything appears to be working now.
Thanks @Haoxiang-Wang and @cest-andre for reporting this!
I've just created a branch called refactor-device
which should help to resolve this e.g. the render_vis
function should automatically detect the appropriate model device.
Will test this and merge to dev if there are no issues.
When using GPU, this codebase only works for
torch.device('cuda:0')
-- the GPU index has to be0
.For example, if you choos
torch.device('cuda:1')
, then when you run the code demoyou will see an error like