Add cudatoolkit
@@ -21,6 +21,7 @@ requirements:
host:
- pip
- python
+ - cudatoolkit
run:
- aim
- contrastive-learner >=0.1.0
@@ -35,15 +36,13 @@ requirements:
- torchvision
- tqdm
- vector-quantize-pytorch >=0.1.0
+ - cudatoolkit
test:
imports:
- stylegan2_pytorch
commands:
- - pip check
- stylegan2_pytorch --help
- requires:
- - pip
about:
home: https://github.com/lucidrains/stylegan2-pytorch
