将n卡的判断条件由nvcc改成nvidia-smi

This commit is contained in:
C3EZ 2025-02-19 22:34:28 +08:00 committed by GitHub
parent f7dc7124a6
commit 3c98f6134f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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