fix

flexflow · Feb 22, 2025 · 426bf43 · 426bf43
1 parent 6ad0754
commit 426bf43
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docker/build.sh b/docker/build.sh
@@ -185,6 +185,7 @@ if [[ "${FF_GPU_BACKEND}" == "hip_rocm" || "${FF_GPU_BACKEND}" == "hip_cuda" ]];
   if [[ "${FF_GPU_BACKEND}" == "hip_rocm" ]]; then
     gpu_backend_version="-${hip_version}"
   fi
+  cuda_version="empty"
 fi
 
 # Get number of cores available on the machine. Build with all cores but one, to prevent RAM choking