CUDA kernel failed : invalid device function

[751565516]

My env is RTX SUPER 2070,CUDA10.1,PYTORCH1.7.0,python3.7.
When I run bash ./scripts/train.sh 0 --config ./cfgs/KITTI_models/PoinTr.yaml --exp_name example
I got a error:
CUDA kernel failed : invalid device function
void furthest_point_sampling_kernel_wrapper(int, int, int, const float*, float*, int*) at L:228 in /tmp/pip-install-1fkhsme8/pointnet2-ops_2dcfc30f04014384afee68e794cd26b7/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/sampling_gpu.cu
./scripts/train.sh: line 8: 2704 Segmentation fault (core dumped) CUDA_VISIBLE_DEVICES=${GPUS} python main.py ${PY_ARGS}
I tried to solve it,but failed.Can you help me?

[yuxumin]

I think the problem may come from your pytorch version or cuda version.Here and here are the same problem. Hope it can help.