search

dmlc / dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.
http://dgl.ai
Apache License 2.0
13.4k stars 3k forks source link

Only CPU versions can be installed by commands in the `Get Started` page with options `Windows` + `CUDA` selected. #7476

Closed chuanwise closed 2 months ago

chuanwise commented 3 months ago

I want to install DGL in my computer with Windows 11 and CUDA 12.1. I selected options PyTorch 2.3.x, CUDA 12.1, Package Pip(Stable) and OS: Windows in this page, and it given me commands:

# If you have installed dgl-cuXX package, please uninstall it first.
pip install  dgl -f https://data.dgl.ai/wheels/cu121/repo.html

After run that, dgl with ONLY CPU version is installed. Users who want DGL CUDA version can only build and install it from source.

Rhett-Ying commented 3 months ago

DGL 2.2.1 + cuda 12.1 is released. please install with below command;

# If you have installed dgl-cuXX package, please uninstall it first.
pip install  dgl -f https://data.dgl.ai/wheels/cu121/repo.html
mfbalin commented 2 months ago

Is this still a problem even with the latest release?