From f71aba8a08b533b085e8ae49acccce7bac8317aa Mon Sep 17 00:00:00 2001 From: David PAGNON Date: Thu, 8 Aug 2024 08:33:01 +0200 Subject: [PATCH] Typo in if torch.cuda.is_available --- Pose2Sim/poseEstimation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Pose2Sim/poseEstimation.py b/Pose2Sim/poseEstimation.py index 589c60f..d9ef11d 100644 --- a/Pose2Sim/poseEstimation.py +++ b/Pose2Sim/poseEstimation.py @@ -360,7 +360,7 @@ def rtm_estimator(config_dict): try: import torch import onnxruntime as ort - if torch.cuda.is_available() == False and 'CUDAExecutionProvider' in ort.get_available_providers(): + if torch.cuda.is_available() and 'CUDAExecutionProvider' in ort.get_available_providers(): device = 'cuda' backend = 'onnxruntime' logging.info(f"\nValid CUDA installation found: using ONNXRuntime backend with GPU.")