forked from open-webui/open-webui
Merge pull request #575 from Mulugruntz/sgiffard/nvidia-gpu-on-wsl2
Enhance GPU detection logic to support WSL2 environments
This commit is contained in:
commit
ddd1a437ae
1 changed files with 2 additions and 2 deletions
|
@ -11,8 +11,8 @@ TICK='\u2713'
|
||||||
|
|
||||||
# Detect GPU driver
|
# Detect GPU driver
|
||||||
get_gpu_driver() {
|
get_gpu_driver() {
|
||||||
# Detect NVIDIA GPUs
|
# Detect NVIDIA GPUs using lspci or nvidia-smi
|
||||||
if lspci | grep -i nvidia >/dev/null; then
|
if lspci | grep -i nvidia >/dev/null || nvidia-smi >/dev/null 2>&1; then
|
||||||
echo "nvidia"
|
echo "nvidia"
|
||||||
return
|
return
|
||||||
fi
|
fi
|
||||||
|
|
Loading…
Reference in a new issue