From 03757632cf099027bd5b5efafa5b207dbf7f8f15 Mon Sep 17 00:00:00 2001 From: Marc-Andre Ferland Date: Sat, 29 Oct 2022 04:33:14 -0400 Subject: [PATCH] Missing .lower() cause CUDA:0 to fail check where cuda:0 works. --- ui/server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ui/server.py b/ui/server.py index a0dc52ef..c722894f 100644 --- a/ui/server.py +++ b/ui/server.py @@ -96,7 +96,7 @@ def setConfig(config): has_first_cuda_device = False for device in gpu_devices: - if not task_manager.is_first_cuda_device(device): continue + if not task_manager.is_first_cuda_device(device.lower()): continue has_first_cuda_device = True break if len(gpu_devices) > 0 and not has_first_cuda_device: