Google Colab issue

[pablo-arantes]

Hi,

I've been using cg2all on Google Colab and it worked well a few months ago. Today I was trying to use and I keep receiving the same error message:

DGL backend not selected or invalid.  Assuming PyTorch for now.
Setting the default backend to "pytorch". You can change it in the ~/.dgl/config.json file or export the DGLBACKEND environment variable.  Valid options are: pytorch, mxnet, tensorflow (all lowercase)
---------------------------------------------------------------------------
OSError                                   Traceback (most recent call last)
[/usr/local/lib/python3.10/site-packages/torch/__init__.py](https://localhost:8080/#) in _load_global_deps()
    171     try:
--> 172         ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
    173     except OSError as err:

10 frames
OSError: /usr/local/lib/python3.10/site-packages/torch/lib/../../nvidia/cublas/lib/libcublas.so.11: undefined symbol: cublasLtGetStatusString, version libcublasLt.so.11

During handling of the above exception, another exception occurred:

OSError                                   Traceback (most recent call last)
[/usr/lib/python3.10/ctypes/__init__.py](https://localhost:8080/#) in __init__(self, name, mode, handle, use_errno, use_last_error, winmode)
    372 
    373         if handle is None:
--> 374             self._handle = _dlopen(self._name, mode)
    375         else:
    376             self._handle = handle

OSError: /usr/local/lib/python3.10/site-packages/nvidia/cublas/lib/libcublas.so.11: undefined symbol: cublasLtGetStatusString, version libcublasLt.so.11

Can you please help me to address this issue?

Thank you.

[huhlim]

I apologize for very late response. The CUDA version that Google Colab uses has been updated to 12, while the previous notebook installed dependencies with CUDA 11. I created a branch cuda-12 and set latest versions for dgl and torch to use with CUDA 12. I did not tested the notebook extensively, so these change may cause some issues.