train error

[qin-yuhao]

return forward_call(*args, **kwargs)

File "/home/qyh/anaconda3/envs/pgsr/lib/python3.9/site-packages/diff_plane_rasterization/init.py", line 229, in forward
return rasterize_gaussians(
File "/home/qyh/anaconda3/envs/pgsr/lib/python3.9/site-packages/diff_plane_rasterization/init.py", line 34, in rasterize_gaussians
return _RasterizeGaussians.apply(
File "/home/qyh/anaconda3/envs/pgsr/lib/python3.9/site-packages/torch/autograd/function.py", line 575, in apply
return super().apply(*args, **kwargs) # type: ignore[misc]
File "/home/qyh/anaconda3/envs/pgsr/lib/python3.9/site-packages/diff_plane_rasterization/init.py", line 100, in forward
num_rendered, color, radii, out_observe, out_all_map, out_plane_depth, geomBuffer, binningBuffer, imgBuffer = _C.rasterize_gaussians(*args)
RuntimeError: CUDA error: invalid configuration argument
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

Training progress: 0%| | 0/30000 [00:00<?, ?it/s]

[WhyxyxIsNotAnAvailableName]

I got same error, have you solved this yet?