2020-09-17 10:56:58 -07:00
|
|
|
ipython
|
2021-11-01 18:29:48 -07:00
|
|
|
|
|
|
|
|
# Needed for Ray Client error message serialization/deserialization.
|
|
|
|
|
tblib
|
|
|
|
|
|
2022-04-21 09:48:43 -07:00
|
|
|
# If you make changes to the torch versions, please also make the corresponding changes to `requirements_dl.txt`!
|
2021-01-27 13:45:30 -08:00
|
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
2021-09-16 20:16:40 -07:00
|
|
|
torch==1.9.0+cu111
|
|
|
|
|
torchvision==0.10.0+cu111
|
2022-04-21 09:48:43 -07:00
|
|
|
|
|
|
|
|
-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'
|
