try setting cuda v with deps
@@ -36,21 +36,14 @@ requirements:
- scipy
- ase
- pytorch
- - pytorch_cluster
- - pytorch_scatter
- - pytorch_sparse
- - pytorch_spline_conv
+ - pytorch_cluster ==latest+${CUDA}
+ - pytorch_scatter ==latest+${CUDA}
+ - pytorch_sparse ==latest+${CUDA}
+ - pytorch_spline_conv ==latest+${CUDA}
test:
- command: cd tests && pytest .
imports:
- - torch
- torch_geometric
- requires:
- - mock
- - pytest
- - pytest-cov
- - pytest-runner
about:
home: https://github.com/rusty1s/pytorch_geometric
