Add cuda-9 install to face plugin installer

- face plugin requires cuda 9.0.
merge-requests/63/head
Moe 2019-05-16 10:53:33 -07:00
parent 5dc18248e3
commit 478c0937df
1 changed files with 13 additions and 6 deletions

View File

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