Add extra for cupy-cuda115 (#571)

explosion · Dec 20, 2021 · 98dfd51 · 98dfd51
1 parent 1b98a5f
commit 98dfd51
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -81,6 +81,8 @@ cuda113 =
     cupy-cuda113>=5.0.0b4
 cuda114 =
     cupy-cuda114>=5.0.0b4
+cuda115 =
+    cupy-cuda115>=5.0.0b4
 datasets =
     ml_datasets>=0.2.0,<0.3.0
 torch =