diff --git a/INSTALL/cuda-10-2.sh b/INSTALL/cuda-10-2.sh index 2f4f6d09..4f9e7c34 100644 --- a/INSTALL/cuda-10-2.sh +++ b/INSTALL/cuda-10-2.sh @@ -12,7 +12,7 @@ if [ -x "$(command -v apt)" ]; then sudo apt-get update -y - sudo apt-get -o Dpkg::Options::="--force-overwrite" install cuda -y --no-install-recommends + sudo apt-get -o Dpkg::Options::="--force-overwrite" install cuda-toolkit-10-2 -y --no-install-recommends sudo apt-get -o Dpkg::Options::="--force-overwrite" install --fix-broken -y # Install CUDA DNN diff --git a/INSTALL/cuda.sh b/INSTALL/cuda.sh index 2f4f6d09..4f9e7c34 100644 --- a/INSTALL/cuda.sh +++ b/INSTALL/cuda.sh @@ -12,7 +12,7 @@ if [ -x "$(command -v apt)" ]; then sudo apt-get update -y - sudo apt-get -o Dpkg::Options::="--force-overwrite" install cuda -y --no-install-recommends + sudo apt-get -o Dpkg::Options::="--force-overwrite" install cuda-toolkit-10-2 -y --no-install-recommends sudo apt-get -o Dpkg::Options::="--force-overwrite" install --fix-broken -y # Install CUDA DNN