How to run it with CUDA capability sm_89?

[zhengyunxuan]

I have a computer with an Nvidia 4090 graphics card, and I want to run the bpreg program on it. However, after installing bpreg, when I use bpreg_predict, I get a warning that the current PyTorch version does not compatible with the sm_89.

UserWarning:
NVIDIA GeForce RTX 4090 with CUDA capability sm_89 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.

Then I want to use cpu only version pytorch to run the program, I install the cpu only version on the official website of pytorch, and when I am installing bpreg, the installer automatically uninstalled the cpu only version pytorch and installed pytorch 1.8.1 with gpu. But I can't use it for this pytorch version is not compatible with sm_89.
I don't know how to solve this problem, I would be grateful if someone could provide assistance :)