try gpu version for pytorch deps
@@ -37,11 +37,10 @@ requirements:
- scikit-learn
- scipy
- ase
- - pytorch
- - pytorch_cluster
- - pytorch_scatter
- - pytorch_sparse
- - pytorch_spline_conv
+ - pytorch_cluster =*=gpu
+ - pytorch_scatter =*=gpu
+ - pytorch_sparse =*=gpu
+ - pytorch_spline_conv =*=gpu
test:
imports:
