update requirement fairscale (no support for pytorch < 1.8.1

shinichiye · Nov 23, 2021 · a31989d · a31989d
1 parent d789f15
commit a31989d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/installers/install_fairscale.sh b/tools/installers/install_fairscale.sh
@@ -38,10 +38,10 @@ if ! "${python_36_plus}"; then
     exit 1
 else
 
-    if $(pt_plus 1.7.0); then
+    if $(pt_plus 1.8.1); then
         pip install fairscale
     else
-        echo "[WARNING] fairscale requires pytorch>=1.6.0"
+        echo "[WARNING] fairscale requires pytorch>=1.8.1"
     fi
 
 fi