Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

version of cuda and dgl #6

Open
julycetc opened this issue Jun 11, 2020 · 1 comment
Open

version of cuda and dgl #6

julycetc opened this issue Jun 11, 2020 · 1 comment

Comments

@julycetc
Copy link

Hello, when i run the code,i get the following error:
Check failed: allow_missing: Device API gpu is not enabled. Please install the cuda version of dgl.
so i wuold like to know what is your version of cuda and dgl? Thanks very much.

@fodothing
Copy link

fodothing commented Feb 10, 2022

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants