anonymous

Maybe this code could help you, good luck! ``` import torch try: import dgl except ModuleNotFoundError: CUDA = 'cu' + torch.version.cuda.replace('.','') !pip install dgl-{CUDA} -f https://data.dgl.ai/wheels/repo.html ```