Does dgl support apex?

Hi, I am wondering does dgl support apex? I have tried to add apex when using dgl, and it always returns some errors. I am wondering does dgl support apex? Thanks!

If you refer to NVIDIA/apex, DGL has not supported it. See also Mixed precision training of GNNs using apex.

Thanks! I am wondering will it be possible to support apex in the further as if the graph is relatively large apex could help with reducing cuda memory and improving computation speed?

Mixed precision training will be officially supported in DGL 0.6.

1 Like