From 94e5288198f1bac1ebf1f99c95c60aea90f3d1b9 Mon Sep 17 00:00:00 2001 From: Colin Date: Wed, 4 Jan 2023 23:01:31 -0500 Subject: [PATCH] fix: Use APT_INSTALL variable consistently --- ml_in_a_box.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ml_in_a_box.sh b/ml_in_a_box.sh index 1364c01..2843b23 100644 --- a/ml_in_a_box.sh +++ b/ml_in_a_box.sh @@ -146,8 +146,8 @@ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" - sudo apt-get install libcudnn8=8.5.0.*-1+cuda11.7 - sudo apt-get install libcudnn8-dev=8.5.0.*-1+cuda11.7 + sudo $APT_INSTALL libcudnn8=8.5.0.*-1+cuda11.7 + sudo $APT_INSTALL libcudnn8-dev=8.5.0.*-1+cuda11.7 # ==================================================================