build.sh: use disable_cuda option

BabitMF · Feb 18, 2024 · 64f5257 · 64f5257
1 parent 27d8dbd
commit 64f5257
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/build.sh b/build.sh
@@ -60,13 +60,9 @@ then
     then
         LOCAL_BUILD=0
     fi
-    if [ "$1" = "disable" ]
+    if [ "$1" = "disable_cuda" ]
     then
-        shift
-        if [ "$1" = "cuda" ]
-        then
-            CUDA_ENABLE=OFF
-        fi
+        CUDA_ENABLE=OFF
     fi
 fi
 

diff --git a/docker/Dockerfile.cpu b/docker/Dockerfile.cpu
@@ -22,7 +22,7 @@ WORKDIR /root/bmf/
 RUN scripts/build_ffmpeg.sh --device=$DEVICE nasm yasm x264 x265 fdk-aac opus \
     && rm -rf ffmpeg_source
 
-RUN ./build.sh disable cuda && rm -rf .git && ./build.sh clean
+RUN ./build.sh disable_cuda && rm -rf .git && ./build.sh clean
 
 ENV LD_LIBRARY_PATH=/root/bmf/output/bmf/lib
 ENV PATH=${PATH}:/root/bmf/output/bmf/bin