Training PV-RCNN-Custom Dataset-CUDA10.1-Pytorch compatibility-ERROR

[HackerHakcer]

Hello friends,
I am trying to install OpenPCDet on Ubuntu 20.04.4 LTS.
And i am running into error when i try to train PV-RCNN on a custom dataset using this command " python3 /home/nawfal/OpenPCDet/tools/train.py --cfg_file /home/nawfal/OpenPCDet/tools/cfgs/custom_models/pv_rcnn.yaml --batch_size 4 --epochs 50"
and i run into this error beIlow.
I think it is a problem of compatibility i want to know the version of pytorch that i need to install for cuda10.1.
If it is not the case of error please correct it to me.
Traceback (most recent call last):
File "/home/nawfal/OpenPCDet/tools/train.py", line 7, in
from test import repeat_eval_ckpt
File "/home/nawfal/OpenPCDet/tools/test.py", line 14, in
from eval_utils import eval_utils
File "/home/nawfal/OpenPCDet/tools/eval_utils/eval_utils.py", line 8, in
from pcdet.models import load_data_to_gpu
File "/home/nawfal/OpenPCDet/pcdet/models/init.py", line 6, in
from .detectors import build_detector
File "/home/nawfal/OpenPCDet/pcdet/models/detectors/init.py", line 1, in
from .detector3d_template import Detector3DTemplate
File "/home/nawfal/OpenPCDet/pcdet/models/detectors/detector3d_template.py", line 6, in
from ...ops.iou3d_nms import iou3d_nms_utils
File "/home/nawfal/OpenPCDet/pcdet/ops/iou3d_nms/iou3d_nms_utils.py", line 9, in
from . import iou3d_nms_cuda
ImportError: /home/nawfal/OpenPCDet/pcdet/ops/iou3d_nms/iou3d_nms_cuda.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZNK2at6Tensor8data_ptrIfEEPT_v