fix meta.yaml build section
@@ -13,11 +13,9 @@ source:
build:
number: 0
noarch: python
+ skip: true # [cuda_compiler_version == "None" or cuda_compiler_version == "10.2" ]
entry_points:
- stylegan2_pytorch = stylegan2_pytorch.cli:main
- script: build.sh
- string: py{{ CONDA_PY }}h{{ PKG_HASH }}_{{ PKG_BUILDNUM }}_{{ torch_proc_type }} # [cuda_compiler_version == "None"]
- string: py{{ CONDA_PY }}cuda{{ cuda_compiler_version|replace('.', '') }}h{{ PKG_HASH }}_{{ PKG_BUILDNUM }}_{{ torch_proc_type }} # [cuda_compiler_version != "None"]
requirements:
@@ -35,7 +33,7 @@ requirements:
- pillow
- python <=3.8
- retry
- - pytorch
+ - pytorch >=1.8.0=cuda*
- torchvision
- tqdm
- vector-quantize-pytorch >=0.1.0
