diff --git a/dependencies/requirements.gpu.txt b/dependencies/requirements.gpu.txt
index 20dcc3e..7f24e0e 100644
--- a/dependencies/requirements.gpu.txt
+++ b/dependencies/requirements.gpu.txt
@@ -21,3 +21,4 @@ skrebate==0.62
 xgboost==2.0.3
 stopit==1.1.2
 tpot==0.12.2
+umap-learn==0.5.6