nv-tlabs / NKSR

[CVPR 2023 Highlight] Neural Kernel Surface Reconstruction
https://research.nvidia.com/labs/toronto-ai/NKSR
Other
735 stars 43 forks source link

RuntimeError: CUDA error: no kernel image is available for execution on the device #57

Open Golriz-code opened 8 months ago

Golriz-code commented 8 months ago

Hello, Thank you for your great work. I have installed the environment, but i got this error during training: ^MSanity Checking: 0it [00:00, ?it/s]^MSanity Checking: 0%| | 0/2 [00:00<?, ?it/s]^MSanity Checking DataLoader 0: 0%| | 0/2 [00:00<?, ?it/s]^[[96m12-18 07:41:02 (base_model.py:155)^[[0m ^[[93m[WARNING] Validation-step Runtime Error. Skipping. ^[[0m Traceback (most recent call last): File "/project/6006041/golriz/nksr/models/base_model.py", line 152, in validation_step return self.train_val_step(is_val=True, *args, kwargs) File "/project/6006041/golriz/nksr/models/nksr_net.py", line 238, in train_val_step out = self(batch, out) File "/project/6006041/golriz/test_nksr/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(*args, *kwargs) File "/project/6006041/golriz/nksr/models/nksr_net.py", line 73, in forward feat = self.network.encoder(input_xyz, feat, enc_svh, 0) File "/project/6006041/golriz/test_nksr/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(args, kwargs) File "/project/6006041/golriz/test_nksr/lib/python3.10/site-packages/nksr/nn/encdec.py", line 89, in forward pts_xyz = grid.world_to_grid(pts_xyz) File "/project/6006041/golriz/test_nksr/lib/python3.10/site-packages/nksr/svh.py", line 147, in world_to_grid return self._grid.world_to_grid(pts) RuntimeError: CUDA error: no kernel image is available for execution on the device Compile with TORCH_USE_CUDA_DSA to enable device-side assertions. I checked the version of my installation and yours , all are compatible. Could you help me with this matter?