cuda compiler v
@@ -13,9 +13,7 @@ source:
build:
number: 0
- script:
- - export TORCH_CUDA_ARCH_LIST="5.0 6.1"
- - {{ PYTHON }} -m pip install . -vv
+ script: {{ PYTHON }} -m pip install . -vv
skip: True # [win]
requirements:
@@ -40,10 +38,10 @@ requirements:
- scipy
- ase
- pytorch
- - pytorch_cluster
- - pytorch_scatter
- - pytorch_sparse
- - pytorch_spline_conv
+ - pytorch_cluster {{ cuda_compiler_version }}
+ - pytorch_scatter {{ cuda_compiler_version }}
+ - pytorch_sparse {{ cuda_compiler_version }}
+ - pytorch_spline_conv {{ cuda_compiler_version }}
test:
imports:
