From 7e68260f712fff0518a4234d42186173395e4879 Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Wed, 31 May 2023 10:38:16 +0530 Subject: [PATCH] bruh moment 11 --- run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run.py b/run.py index a9570fc..fe74a2c 100644 --- a/run.py +++ b/run.py @@ -68,7 +68,7 @@ def pre_check(): CUDA_VERSION = torch.version.cuda CUDNN_VERSION = torch.backends.cudnn.version() NVIDIA_PROVIDERS = ['CUDAExecutionProvider', 'TensorrtExecutionProvider'] - if list(set(core.globals.providers) - set(NVIDIA_PROVIDERS)) > 1: + if len(list(set(core.globals.providers) - set(NVIDIA_PROVIDERS))) > 1: if not torch.cuda.is_available() or not CUDA_VERSION: quit("You are using --gpu flag but CUDA isn't available or properly installed on your system.") if CUDA_VERSION > '11.8':