Add DGL to requirements file

Questions
#1

I am using DGL in databricks. In order to serve a model Ineed to specify the correct DGL version otherwise it will install the cpu version.
Since the version of dgl to install depends on the version of cuda, I am having a hard time figuring out a way to do this.
Is there a way to install the appropriate version of dgl via requirements.txt?

#2

I think you can do
dgl @ https://data.dgl.ai/wheels/torch-2.4/cu121/repo.html

Or use --find-links format.

Or whatever link that matches your version requirements from this page: Deep Graph Library

#3

The --find-links format worked fine, ty :slight_smile:

#4

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