Hotfix installer
This commit is contained in:
parent
2bc65fe576
commit
283deb07fa
@ -70,7 +70,7 @@ def run(program : ArgumentParser) -> None:
|
||||
os.remove(wheel_path)
|
||||
else:
|
||||
subprocess.call([ 'pip', 'uninstall', 'onnxruntime', onnxruntime_name, '-y', '-q' ])
|
||||
if onnxruntime == 'cuda-12.1':
|
||||
if onnxruntime == 'cuda-12.2':
|
||||
subprocess.call([ 'pip', 'install', onnxruntime_name + '==' + onnxruntime_version, '--extra-index-url', 'https://pkgs.dev.azure.com/onnxruntime/onnxruntime/_packaging/onnxruntime-cuda-12/pypi/simple', '--force-reinstall' ])
|
||||
else:
|
||||
subprocess.call([ 'pip', 'install', onnxruntime_name + '==' + onnxruntime_version, '--force-reinstall' ])
|
||||
|
Loading…
Reference in New Issue
Block a user