The detected CUDA version (10.1) mismatches the version that was used to compile PyTorch (11.3).

[LZ-CH]

Hi,

My environment is:

Ubuntu 18.04
Python 3.9.0
PyTorch 1.12.0 with CUDA 11.3
GeForce RTX 3090 GPU (Unable to use cuda 10.1)

When I install pointnet2 with CUDA 11.3, I encounter the following error:

File "/data1/user/anaconda3/envs/ScanNet/lib/python3.9/site-packages/torch/utils/cpp_extension.py", line 812, in _check_cuda_version
    raise RuntimeError(CUDA_MISMATCH_MESSAGE.format(cuda_str_version, torch.version.cuda))
RuntimeError: 
The detected CUDA version (10.1) mismatches the version that was used to compile
PyTorch (11.3). Please make sure to use the same CUDA versions.

And, How do I solve this problem?

Thanks!