diff --git a/Project.toml b/Project.toml
index db885e2a..052f61ad 100644
--- a/Project.toml
+++ b/Project.toml
@@ -28,7 +28,7 @@ AbstractTrees = "0.2, 0.3"
 Adapt = "1"
 CodecZlib = "0.5, 0.6, 0.7"
 Colors = "0.8, 0.9, 0.10, 0.11, 0.12"
-CuArrays = "1.6"
+CuArrays = "1.6, 2"
 Juno = "0.5, 0.6, 0.7, 0.8"
 MacroTools = "0.3, 0.4, 0.5"
 NNlib = "0.6"