fix

flexflow · Mar 3, 2024 · 71d296b · 71d296b
1 parent 7f0831f
commit 71d296b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/pull.sh b/docker/pull.sh
@@ -45,8 +45,8 @@ if [[ "${FF_GPU_BACKEND}" == "cuda" || "${FF_GPU_BACKEND}" == "hip_cuda" ]]; the
     fi
   fi
   # Check that CUDA version is supported
-  if [[ "$cuda_version" != @(11.1|11.3|11.6|11.8|12.0|12.1|12.2) ]]; then
-    echo "cuda_version is not available for download, please choose among {11.1|11.3|11.6|11.8|12.0|12.1|12.2}"
+  if [[ "$cuda_version" != @(11.1|11.6|11.7|11.8|12.0|12.1|12.2) ]]; then
+    echo "cuda_version is not available for download, please choose among {11.1|11.6|11.7|11.8|12.0|12.1|12.2}"
     exit 1
   fi
   # Use CUDA 12.2 for all versions greater or equal to 12.2 for now