Updated start.sh to enable OpenCL

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

View file

@ -1,5 +1,7 @@
#!/bin/bash #!/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/ DATA_FOLDER=$(pwd)/data/
declare CAM0 CAM1 CAM2 CAM3 declare CAM0 CAM1 CAM2 CAM3
printf "\n" printf "\n"
@ -28,6 +30,6 @@ shift "$(($OPTIND -1))"
printf "\n" printf "\n"
# Warning xhost + is overly permissive and will reduce system security. Edit as desired # 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 + 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 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