This commit is contained in:
Somdev Sangwan 2023-05-30 08:52:21 +05:30 committed by GitHub
parent 997eda5695
commit de9bf62d0c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

2
run.py
View File

@ -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