🚑 fix(deps): hotfix version cupy-cuda11x==11.0.0 (#17)

KiseKloset · Apr 20, 2024 · 5dcd04e · 5dcd04e
1 parent 8dc7048
commit 5dcd04e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 # Model
 torch>=1.13.1
 torchvision>=0.14.1
-cupy-cuda11x>=11.0.0
+cupy-cuda11x==11.0.0
 
 # Monitor/Metric
 tensorboardX>=2.6