This commit is contained in:
Somdev Sangwan
2023-05-30 08:52:21 +05:30
committed by GitHub
parent 997eda5695
commit de9bf62d0c

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