Updated start.sh to enable OpenCL

iperov · Oct 1, 2022 · c2e26d5 · c2e26d5
1 parent 05c9fae
commit c2e26d5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/build/linux/start.sh b/build/linux/start.sh
@@ -1,5 +1,7 @@
 #!/bin/bash
 
+NV_VER=$(modinfo /usr/lib/modules/$(uname -r)/updates/dkms/nvidia.ko | grep ^version |awk '{print $2}'|awk -F '.' '{print $1}')
+
 DATA_FOLDER=$(pwd)/data/
 declare CAM0 CAM1 CAM2 CAM3
 printf "\n"
@@ -28,6 +30,6 @@ shift "$(($OPTIND -1))"
 printf "\n"
 
 # Warning xhost + is overly permissive and will reduce system security. Edit as desired
-docker build . -t deepfacelive
+docker build . -t deepfacelive --build-arg NV_VER=$NV_VER
 xhost +
 docker run --ipc host --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix -v $DATA_FOLDER:/data/ $CAM0 $CAM1 $CAM2 $CAM3  --rm -it deepfacelive