rapidsai / wholegraph

WholeGraph - large scale Graph Neural Networks
https://docs.rapids.ai/api/cugraph/stable/wholegraph/
Apache License 2.0
100 stars 38 forks source link

Handling of PyTorch dependency #210

Open jakirkham opened 3 months ago

jakirkham commented 3 months ago

During a recent review of dependencies here we noticed that PyTorch is a dependency, but is not listed as one at runtime

https://github.com/rapidsai/wholegraph/blob/9b877a45519d3aba707884fddf9c3a837ac32719/conda/recipes/pylibwholegraph/meta.yaml#L68-L74

However there are a range of tightly pinned PyTorch builds (on the older side). All of them for CUDA 11. No CUDA 12

https://github.com/rapidsai/wholegraph/blob/9b877a45519d3aba707884fddf9c3a837ac32719/dependencies.yaml#L230-L291

Not really sure how to handle updating PyTorch here or how it should be reflected in dependencies. So opening this issue to discuss and find a reasonable resolution