Fix yolo cuda check

merge-requests/61/head
Moe 2019-04-27 11:55:26 -07:00
parent f5d416bf54
commit 5c5e5a66f6
1 changed files with 1 additions and 5 deletions

View File

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