meta.yaml 879 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
package:
  name: pytorch-cluster
  version: 1.5.9

source:
  path: ../..

requirements:
  build:
    - {{ compiler('c') }}  # [win]

  host:
    - python {{ environ.get('PYTHON_VERSION') }}
    - {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
    - {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}

  run:
    - python {{ environ.get('PYTHON_VERSION') }}
    - {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
    - {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}

build:
  string: py{{ environ.get('PYTHON_VERSION').replace('.', '') }}_torch_{{ environ['TORCH_VERSION'] }}_{{ environ['CUDA_VERSION'] }}
  script: python setup.py install
  script_env:
    - FORCE_CUDA

test:
  imports:
    - torch_cluster
rusty1s's avatar
rusty1s committed
31
32
  requires:
    - scipy
rusty1s's avatar
rusty1s committed
33
34
35
36
37

about:
  home: https://github.com/rusty1s/pytorch_cluster
  license: MIT
  summary: PyTorch Extension Library of Optimized Graph Cluster Algorithms