MegEngine / FINet

This is the official MegEngine implementation of FINet: Dual Branches Feature Interaction for Partial-to-Partial Point Cloud Registration, AAAI 2022
MIT License
24 stars 1 forks source link

there is a conflict between cuda and megengine #5

Open Qig666 opened 8 months ago

Qig666 commented 8 months ago

We have met the problem when we train FINet, do you have any idea? Thanks.

Traceback (most recent call last): File "/home/hongchang/.conda/envs/finet/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/home/hongchang/.conda/envs/finet/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/home/hongchang/.conda/envs/finet/lib/python3.8/site-packages/megengine/distributed/launcher.py", line 43, in _run_wrapped _check_device_initialized(device_type, dev) File "/home/hongchang/.conda/envs/finet/lib/python3.8/site-packages/megengine/distributed/helper.py", line 195, in _check_device_initialized raise RuntimeError(errmsg) RuntimeError: The cuda env is set before the forked thread starts. Please do not use any cuda function or variable before forking. Process Process-2: