Introduce pre_check(), Cleanup requirements.txt
This commit is contained in:
parent
365f3d790a
commit
fe9d6b9cae
2
run.py
2
run.py
@ -52,7 +52,7 @@ def pre_check():
|
||||
if 'ROCMExecutionProvider' not in core.globals.providers:
|
||||
if CUDA_VERSION > '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.")
|
||||
if CUDNN_VERSION < 8220:
|
||||
quit(f"CUDNN version {CUDNN_VERSION} is not supported - please upgrade to 8.9.1")
|
||||
|
Loading…
Reference in New Issue
Block a user