From 3c98f6134f5e4523205b571853dc54fd14f3db3e Mon Sep 17 00:00:00 2001 From: C3EZ <96614352+luckykevvv@users.noreply.github.com> Date: Wed, 19 Feb 2025 22:34:28 +0800 Subject: [PATCH] =?UTF-8?q?=E5=B0=86n=E5=8D=A1=E7=9A=84=E5=88=A4=E6=96=AD?= =?UTF-8?q?=E6=9D=A1=E4=BB=B6=E7=94=B1nvcc=E6=94=B9=E6=88=90nvidia-smi?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- install.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/install.sh b/install.sh index 7b5fca4..1fdd68b 100644 --- a/install.sh +++ b/install.sh @@ -9,7 +9,7 @@ echo "Installing ffmpeg and cmake..." conda install ffmpeg cmake echo "Checking for CUDA installation..." -if command -v nvcc &> /dev/null; then +if command -v nvidia-smi &> /dev/null; then USE_CUDA=true echo "CUDA found." else