diff --git a/environment.yml b/environment.yml index c28003d146cba7df7ba03864ac212c5cfc23343e..45e0ead73a7f29a047428e6e041e9d331580b616 100644 --- a/environment.yml +++ b/environment.yml @@ -4,11 +4,11 @@ channels: - rusty1s dependencies: - python - - pytorch - pip - pip: - --find-links https://data.pyg.org/whl/torch-1.9.0+cu102.html - scikit-learn + - torch - torch-scatter - torch-sparse - torch-cluster