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

[DiscipleK]

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

[BarclayII]

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

[DiscipleK]

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