Running Error.

[KisAaki]

Hi, when running this code, I met a problem when I'm trying to run it on 2 GPU.How can I fix this problem?
File "/data1/anaconda3/envs/yzy_pgst/lib/python3.9/site-packages/torch/nn/parallel/parallel_apply.py", line 64, in _worker
output = module(*input, **kwargs)
File "/data1/anaconda3/envs/yzy_pgst/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/data2/2023/yzy/models/PointGPT.py", line 778, in forward
_, U = torch.linalg.eigh(L1)
RuntimeError: lazy wrapper should be called at most once

[jerryfeng2003]

Hi,
Adding the code torch.cuda.current_device() before _, U = torch.linalg.eigh(L1) may solve the problem.