However, in torchpq-0.3.0.1, there is a bug that they always create new tensors on cuda:0, it is a problem for those who have multiple devices. The author of torchpq fixed the bug since version 0.3.0.3 (see link).
So I suggest you update the install requirements in setup.py to
Hi, @minyoungg . Thanks for the great work. I'd like to point out one issue which may result in CUDA device issue.
Your install requirements are
in
setup.py
.However, in
torchpq-0.3.0.1
, there is a bug that they always create new tensors oncuda:0
, it is a problem for those who have multiple devices. The author oftorchpq
fixed the bug since version 0.3.0.3 (see link).So I suggest you update the install requirements in
setup.py
toBest, Chris :)