search

Nerogar / OneTrainer

OneTrainer is a one-stop solution for all your stable diffusion training needs.
GNU Affero General Public License v3.0
1.81k stars 154 forks source link

Auto-detect NVIDIA GPUs inside Microsoft WSL #534

Closed Arcitec closed 3 weeks ago

Arcitec commented 3 weeks ago

NVIDIA uses a non-standard path for their WSL driver, which is documented here:

https://docs.nvidia.com/cuda/wsl-user-guide/

We now detect that special path too, so that OneTrainer's WSL users will automatically receive the NVIDIA CUDA dependencies.

This patch also removes the old "nvcc" check, which wasn't helpful, since it's not installed by default in NVIDIA drivers, and some people might have the CUDA SDK installed for software building purposes without having an NVIDIA GPU themselves.

Arcitec commented 3 weeks ago

Solution has been verified by @djp3k05 here:

https://github.com/Nerogar/OneTrainer/issues/527#issuecomment-2452678351