Allow_missing: Device API cuda is not enabled. Please install the cuda version of dgl

Check failed: allow_missing: Device API cuda is not enabled. Please install the cuda version of dgl.

dgl.version = 2.0.0
torch.cuda.device_count() = 1
torch.cuda.is_available() = True
torch.version = 2.1.0+cu121
I double check that I install the dgl with command:
pip install dgl -f https://data.dgl.ai/wheels/cu121/repo.html
pip install dglgo -f https://data.dgl.ai/wheels-test/repo.html

What is your system? Windows support for CUDA 12.1 has not been released yet.

It’s windows. So CUDA 11.7 or 11.8 and corrsponding pytorch may work? Thanks for your reply.