tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

Merge pull request #575 from Mulugruntz/sgiffard/nvidia-gpu-on-wsl2

Browse source 
Enhance GPU detection logic to support WSL2 environments
This commit is contained in:
Timothy Jaeryang Baek 2024-01-25 10:40:27 -08:00 • committed by GitHub
commit ddd1a437ae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
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 # 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