Installation with pytorch cuda

Hi, I am trying to install DGL. I am using virtualenv, python 3.8. I installed dgl using

pip install  dgl -f https://data.dgl.ai/wheels/torch-2.3/cu121/repo.html

I have

torch.__version__
'2.3.0+cu121'

However, I run into

>>> import dgl
Traceback (most recent call last):
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/dgl/graphbolt/__init__.py", line 31, in load_graphbolt
    torch.classes.load_library(path)
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/torch/_classes.py", line 51, in load_library
    torch.ops.load_library(path)
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/torch/_ops.py", line 1032, in load_library
    ctypes.CDLL(path)
  File "/usr/lib/python3.8/ctypes/__init__.py", line 373, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: libnvrtc.so.12: cannot open shared object file: No such file or directory

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/dgl/__init__.py", line 16, in <module>
    from . import (
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/dgl/dataloading/__init__.py", line 13, in <module>
    from .dataloader import *
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/dgl/dataloading/dataloader.py", line 27, in <module>
    from ..distributed import DistGraph
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/dgl/distributed/__init__.py", line 5, in <module>
    from .dist_graph import DistGraph, DistGraphServer, edge_split, node_split
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/dgl/distributed/dist_graph.py", line 12, in <module>
    from .. import backend as F, graphbolt as gb, heterograph_index
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/dgl/graphbolt/__init__.py", line 36, in <module>
    load_graphbolt()
  File "/nfs/c3po/home/ge78muc/modulus_stokes_demo/venv/lib/python3.8/site-packages/dgl/graphbolt/__init__.py", line 33, in load_graphbolt
    raise ImportError("Cannot load Graphbolt C++ library")
ImportError: Cannot load Graphbolt C++ library

Any ideas? I already played around with different versions of dgl, pytorch, and deleted and recreated the whole venv. I do not care about software versions, I just want any version running.

You are using the DLG with CUDA 12.1. Have you installed CUDA v12.1 on your machine and exported LD_LIBRARY_PATH?

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