Add DGL to requirements file

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?