pytorch sparse 0.6.1
{% set name = "torch_geometric" %}
{% set version = "1.5.0" %}
-{% set proc_type = "cpu" if cuda_compiler_version == "None" else "gpu" %}
-
package:
name: py{{ name|lower }}
version: {{ version }}
@@ -22,7 +20,6 @@ requirements:
- pytest-runner
- python
- setuptools
- - pytorch
run:
- python
- googledrivedownloader
@@ -38,9 +35,10 @@ requirements:
- scikit-learn
- scipy
- ase
+ - pytorch
- pytorch_cluster
- pytorch_scatter
- - pytorch_sparse
+ - pytorch_sparse == 0.6.1
- pytorch_spline_conv
test:
