diff --git a/Project.toml b/Project.toml
index 462fd8bf..23b98afb 100644
--- a/Project.toml
+++ b/Project.toml
@@ -18,6 +18,7 @@ Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
 [compat]
 CategoricalArrays = "0.10"
 ColorTypes = "0.10.3, 0.11"
+CUDA = "4"
 ComputationalResources = "0.3.2"
 Flux = "0.13, 0.14"
 MLJModelInterface = "1.1.1"