How can I use DGL with cuda

Questions
#1

Hello.
I’m trying to use dgl with gpu.
My computer is Window 10 and has 4 GPU (Tesla V100) and its torch version is 1.12.1 + cu102 and cuda version is 10.2.
I installed dgl and dgl-cu102 from pypi but what i can use is only dgl with cpu.
How can I use gpu? Uninstall dgl cpu version? Upgrade Cuda version?
Thanks.

#2

pip uninstall dgl for uninstall first. Then pip install dgl -f https://data.dgl.ai/wheels/cu102/repo.html.
See Deep Graph Library.

#3

This installs the newer versions of DGL. How can you install, for example 0.9.0 with CUDA?

#4

Install DGL previous version is the duplicate question. please refer to this page for discussion.

#5

This topic was automatically closed 30 days after the last reply. New replies are no longer allowed.