diff --git a/gui.sh b/gui.sh index a5635cd..ffa75fd 100755 --- a/gui.sh +++ b/gui.sh @@ -81,6 +81,8 @@ else if [ "$RUNPOD" = false ]; then if [[ "$@" == *"--use-ipex"* ]]; then REQUIREMENTS_FILE="$SCRIPT_DIR/requirements_linux_ipex.txt" + elif [ -x "$(command -v nvidia-smi)" ]; then + REQUIREMENTS_FILE="$SCRIPT_DIR/requirements_linux.txt" elif [[ "$@" == *"--use-rocm"* ]] || [ -x "$(command -v rocminfo)" ] || [ -f "/opt/rocm/bin/rocminfo" ]; then REQUIREMENTS_FILE="$SCRIPT_DIR/requirements_linux_rocm.txt" else diff --git a/setup.sh b/setup.sh index 432b18a..1b35b66 100755 --- a/setup.sh +++ b/setup.sh @@ -218,6 +218,8 @@ install_python_dependencies() { python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_runpod.txt $QUIET elif [ "$USE_IPEX" = true ]; then python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_linux_ipex.txt $QUIET + elif [ -x "$(command -v nvidia-smi)" ]; then + python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_linux.txt $QUIET elif [ "$USE_ROCM" = true ] || [ -x "$(command -v rocminfo)" ] || [ -f "/opt/rocm/bin/rocminfo" ]; then echo "Upgrading pip for ROCm." pip install --upgrade pip # PyTorch ROCm is too large to install with older pip