From de9bf62d0c7e562e28d194ee990517f28496e6e9 Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Tue, 30 May 2023 08:52:21 +0530 Subject: [PATCH] clarity --- run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run.py b/run.py index 38ad9ff..7bd4417 100755 --- a/run.py +++ b/run.py @@ -12,7 +12,7 @@ if '--gpu' not in sys.argv: elif 'ROCMExecutionProvider' not in core.globals.providers: import torch if not torch.cuda.is_available(): - quit("You are using --gpu flag but CUDA isn't available on your system.") + quit("You are using --gpu flag but CUDA isn't available or properly installed on your system.") import glob import argparse