mirror of https://github.com/ollama/ollama
3f30836734
We now do a deeper probe of CUDA devices to verify the library version has the correct compute capability coverage for the device. Due to ROCm also interpreting the CUDA env var to filter AMD devices, we try to avoid setting it which leads to problems in mixed vendor systems. However without setting it for this deeper probe, each CUDA library subprocess discovers all CUDA GPUs and on systems with lots of GPUs, this can lead to hitting timeouts. The fix is to turn on the CUDA visibility env var just for this deeper probe use-case. |
||
|---|---|---|
| .. | ||
| cpu_linux.go | ||
| cpu_linux_test.go | ||
| cpu_windows.go | ||
| cpu_windows_test.go | ||
| gpu.go | ||
| gpu_darwin.go | ||
| gpu_info_darwin.h | ||
| gpu_info_darwin.m | ||
| runner.go | ||
| runner_test.go | ||
| types.go | ||