Updated start.sh to enable OpenCL

This commit is contained in:
chris 2022-09-30 23:10:02 -04:00
parent 05c9faef36
commit c2e26d5bb7

View file

@ -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