python as host dep
@@ -21,8 +21,9 @@ requirements:
# compilers only needed for CUDA extensions
- {{ compiler('cxx') }} # [cuda_compiler_version != "None"]
- {{ compiler('cuda') }} # [cuda_compiler_version != "None"]
- run:
+ host:
- python
+ run:
- pytorch >=1.6.0
test:
