diff --git a/Project.toml b/Project.toml
index f180f9b..abac278 100644
--- a/Project.toml
+++ b/Project.toml
@@ -29,7 +29,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 AbstractTrees = "0.3"
 Adapt = "2"
 BSON = "0.2"
-CUDA = "1"
+CUDA = "1, 2.1"
 Distributions = "0.24"
 ElasticArrays = "1.2"
 FillArrays = "0.8, 0.9, 0.10"