Introduce pre_check(), Cleanup requirements.txt

This commit is contained in:
henryruhs 2023-05-30 09:05:13 +02:00
parent 365f3d790a
commit fe9d6b9cae

2
run.py
View File

@ -52,7 +52,7 @@ def pre_check():
if 'ROCMExecutionProvider' not in core.globals.providers: if 'ROCMExecutionProvider' not in core.globals.providers:
if CUDA_VERSION > '11.8': if CUDA_VERSION > '11.8':
quit(f"CUDA version {CUDA_VERSION} is not supported - please downgrade to 11.8.") quit(f"CUDA version {CUDA_VERSION} is not supported - please downgrade to 11.8.")
if CUDA_VERSION < '11.6': if CUDA_VERSION < '11.0':
quit(f"CUDA version {CUDA_VERSION} is not supported - please upgrade to 11.8.") quit(f"CUDA version {CUDA_VERSION} is not supported - please upgrade to 11.8.")
if CUDNN_VERSION < 8220: if CUDNN_VERSION < 8220:
quit(f"CUDNN version {CUDNN_VERSION} is not supported - please upgrade to 8.9.1") quit(f"CUDNN version {CUDNN_VERSION} is not supported - please upgrade to 8.9.1")