try cudatoolkit-dev
@@ -22,6 +22,7 @@ requirements:
- pip
- python <=3.8
- cudatoolkit
+ - cudatoolkit-dev
run:
- aim
- contrastive-learner >=0.1.0
@@ -37,6 +38,7 @@ requirements:
- tqdm
- vector-quantize-pytorch >=0.1.0
- cudatoolkit
+ - cudatoolkit-dev
test:
imports:
