mirror of
https://github.com/vale981/ray
synced 2025-03-05 10:01:43 -05:00
732175e245
Add example for distributed pytorch geometric (graph learning) with Ray AIR This only showcases distributed training, but with data small enough that it can be loaded in by each training worker individually. Distributed data ingest is out of scope for this PR. Co-authored-by: matthewdeng <matthew.j.deng@gmail.com>
17 lines
564 B
Text
17 lines
564 B
Text
ipython
|
|
|
|
# Needed for Ray Client error message serialization/deserialization.
|
|
tblib
|
|
|
|
# If you make changes to the torch versions, please also make the corresponding changes to `requirements_dl.txt`!
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
torch==1.9.0+cu111
|
|
torchvision==0.10.0+cu111
|
|
|
|
-f https://data.pyg.org/whl/torch-1.9.0+cu111.html
|
|
torch-scatter==2.0.9
|
|
torch-sparse==0.6.12
|
|
|
|
# torch-geometric has to be installed after torch-scatter and torch-sparse.
|
|
torch-geometric==2.0.3; python_version < '3.7'
|
|
torch-geometric==2.0.4; python_version >= '3.7'
|