try run_constrained
{% set name = "torch_geometric" %}
{% set version = "1.5.0" %}
+{% set cuda_version = environ.get('CUDA_VERSION') %}
+
package:
name: py{{ name|lower }}
version: {{ version }}
@@ -37,10 +39,12 @@ requirements:
- scikit-learn
- scipy
- ase
- - pytorch_cluster =*=gpu
- - pytorch_scatter =*=gpu
- - pytorch_sparse =*=gpu
- - pytorch_spline_conv =*=gpu
+ - pytorch
+ run_constrained:
+ - pytorch_cluster * {{ CUDA_VER }}
+ - pytorch_scatter * {{ CUDA_VER }}
+ - pytorch_sparse * {{ CUDA_VER }}
+ - pytorch_spline_conv * {{ CUDA_VER }}
test:
imports:
