diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 7047383..8bbc5b4 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -91,6 +91,7 @@ zy:
     - source venv/bin/activate
     - uv pip install -r requirements.txt --default-index https://pypi.tuna.tsinghua.edu.cn/simple
   script:
+    - source venv/bin/activate
     - make fetch-quick
     - make sbs-quick
     - make train-quick
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 37234a1..d5acc0a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -19,6 +19,8 @@ matplotlib
 scikit-learn
 scipy
 seaborn
+torch
+torchvision
 kornia
 opencv-python
 opencv-contrib-python