diff --git a/Project.toml b/Project.toml
index 4b171f2..30fe75d 100644
--- a/Project.toml
+++ b/Project.toml
@@ -29,7 +29,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
 AverageShiftedHistograms = "0.8"
-CUDA = "4"
+CUDA = "4, 5"
 ComputationalResources = "0.3"
 DataFrames = "1.2"
 DiffEqFlux = "1.53, 2"