UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 05:24:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Enhance GPU detection logic to support WSL2 environments

Browse Source 
This commit updates the GPU detection logic in the setup script to accommodate environments where NVIDIA GPUs might not be directly listed by `lspci`, such as in Windows Subsystem for Linux 2 (WSL2). The revised logic now attempts to identify NVIDIA GPUs using both `lspci` and `nvidia-smi`, ensuring broader compatibility and recognizing GPUs that are accessible in WSL2. This change allows the script to correctly detect NVIDIA GPUs in a wider range of setups, improving user experience and script robustness in diverse environments.
This commit is contained in:
Samuel GIFFARD 2024-01-25 18:09:53 +01:00 committed by GitHub
parent 0ab33bd5f3
commit 2cb4112145
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
run-compose.sh
View File

@ -11,8 +11,8 @@ TICK='\u2713'
# Detect GPU driver
get_gpu_driver() {
# Detect NVIDIA GPUs
if lspci | grep -i nvidia >/dev/null; then
# Detect NVIDIA GPUs using lspci or nvidia-smi
if lspci | grep -i nvidia >/dev/null || nvidia-smi >/dev/null 2>&1; then
echo "nvidia"
return
fi