diff --git a/train/requirements.txt b/train/requirements.txt
index a702f17..c97f891 100644
--- a/train/requirements.txt
+++ b/train/requirements.txt
@@ -2,4 +2,5 @@ git+https://github.com/supervisely/supervisely.git@semsegm-bm-comparison
 pycocotools
 scikit-learn
 plotly==5.22.0
-kaleido==0.2.1
\ No newline at end of file
+kaleido==0.2.1
+cupy-cuda111
\ No newline at end of file