12345678910111213141516 |
- ipython
- # Needed for Ray Client error message serialization/deserialization.
- tblib
- # If you make changes below this line, please also make the corresponding changes to `requirements_dl.txt`!
- --extra-index-url https://download.pytorch.org/whl/cu116 # for GPU versions of torch, torchvision
- --find-links https://data.pyg.org/whl/torch-1.13.0+cu116.html # for GPU versions of torch-scatter, torch-sparse, torch-cluster, torch-spline-conv
- # specifying explicit plus-notation below so pip overwrites the existing cpu verisons
- torch==1.13.0+cu116
- torchvision==0.14.0+cu116
- torch-scatter==2.1.0+pt113cu116
- torch-sparse==0.6.15+pt113cu116
- torch-cluster==1.6.0+pt113cu116
- torch-spline-conv==1.2.1+pt113cu116
|