specify pytorch cpu vs. cuda depending on build
@@ -27,9 +27,12 @@ requirements:
# upstream uses >=1.6, but there are some issues under py3.9
# with the 1.7 builds from conda-forge
- pytorch >=1.8.0 # [cuda_compiler_version != "None"]
+ - pytorch-gpu # [cuda_compiler_version != "None"]
run:
- python
- pytorch >=1.8.0
+ - pytorch-gpu # [cuda_compiler_version != "None"]
+ - pytorch-cpu # [cuda_compiler_version == "None"]
test:
requires:
