fix: fixed mmcv not being able to install

Seeed-Studio · Nov 13, 2023 · 6a5554e · 6a5554e
1 parent ebb1ec2
commit 6a5554e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/setup_colab.sh b/scripts/setup_colab.sh
@@ -21,6 +21,8 @@ else
     echo -en "${GREEN}OK${RST}\n"
 fi
 
+# limit torch version
+pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1
 
 # install base deps
 echo -en "Installing base deps... "