use pytorch >=1.8.0
@@ -24,10 +24,12 @@ requirements:
host:
- python
- pip
- - pytorch >=1.6.0 # [cuda_compiler_version != "None"]
+ # 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"]
run:
- python
- - pytorch >=1.6.0
+ - pytorch >=1.8.0
test:
requires:
