From 5c5e5a66f6feea132a44a4f5ac969673f7a07cf6 Mon Sep 17 00:00:00 2001 From: Moe Date: Sat, 27 Apr 2019 11:55:26 -0700 Subject: [PATCH] Fix yolo cuda check --- plugins/yolo/INSTALL.sh | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/plugins/yolo/INSTALL.sh b/plugins/yolo/INSTALL.sh index bc0312b5..b12b2c0b 100644 --- a/plugins/yolo/INSTALL.sh +++ b/plugins/yolo/INSTALL.sh @@ -12,9 +12,7 @@ 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 "=============" +if [ -d "/usr/local/cuda" ]; then echo "Do you want to install the plugin with CUDA support?" echo "Do this if you installed NVIDIA Drivers, CUDA Toolkit, and CuDNN" echo "(y)es or (N)o" @@ -23,8 +21,6 @@ if [ ! -d "/usr/local/cuda" ]; then export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH fi -else - echo "CUDA Toolkit found..." fi # echo "-----------------------------------" # if ! [ -x "$(command -v opencv_version)" ]; then