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.")