From 478c0937df894c4b4f9e1e0fe058a23d1cef3b2c Mon Sep 17 00:00:00 2001 From: Moe Date: Thu, 16 May 2019 10:53:33 -0700 Subject: [PATCH] Add cuda-9 install to face plugin installer - face plugin requires cuda 9.0. --- plugins/face/INSTALL.sh | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/plugins/face/INSTALL.sh b/plugins/face/INSTALL.sh index e72720eb..208c395d 100644 --- a/plugins/face/INSTALL.sh +++ b/plugins/face/INSTALL.sh @@ -14,13 +14,20 @@ else echo "$(nvidia-smi |grep 'Driver Version')" fi echo "-----------------------------------" -if [ ! -d "/usr/local/cuda" ]; then - echo "You need to install CUDA Toolkit to use this." - echo "inside the Shinobi directory run the following :" - echo "sh INSTALL/cuda.sh" - exit 1 +if [ ! -d "/usr/local/cuda-9.0" ]; then + echo "Tensorflow requires CUDA Toolkit 9.0" + echo "Installing CUDA Toolki 9.0..." + wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64-deb -O cuda.deb + sudo dpkg -i cuda.deb + sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1704/x86_64/7fa2af80.pub + sudo apt update -y + sudo apt install cuda-toolkit-9-0 -y + wget https://cdn.shinobi.video/installers/libcudnn7_7.5.1.10-1+cuda9.0_amd64.deb -O cuda-dnn.deb + sudo dpkg -i cuda-dnn.deb + wget https://cdn.shinobi.video/installers/libcudnn7-dev_7.5.1.10-1+cuda9.0_amd64.deb -O cuda-dnn-dev.deb + sudo dpkg -i cuda-dnn-dev.deb else - echo "CUDA Toolkit found..." + echo "CUDA Toolkit 9.0 found..." fi echo "-----------------------------------" if [ ! -d "./faces" ]; then