set proc_type
{% set name = "torch_geometric" %}
{% set version = "1.5.0" %}
-{% set cuda_version = environ.get('CUDA_VERSION') %}
+{% set proc_type = "cpu" if cuda_compiler_version == "None" else "gpu" %}
package:
name: py{{ name|lower }}
@@ -13,9 +13,7 @@ source:
build:
number: 0
- script:
- - export FORCE_CUDA="1"
- - {{ PYTHON }} -m pip install . -vv
+ script: {{ PYTHON }} -m pip install . -vv
skip: True # [win]
requirements:
@@ -41,10 +39,10 @@ requirements:
- ase
- pytorch
run_constrained:
- - pytorch_cluster * {{ cuda_version }}
- - pytorch_scatter * {{ cuda_version }}
- - pytorch_sparse * {{ cuda_version }}
- - pytorch_spline_conv * {{ cuda_version }}
+ - pytorch_cluster * {{ proc_type }}
+ - pytorch_scatter * {{ proc_type }}
+ - pytorch_sparse * {{ proc_type }}
+ - pytorch_spline_conv * {{ proc_type }}
test:
imports:
