Hotfix ROCm
This commit is contained in:
@@ -89,5 +89,5 @@ def run(program : ArgumentParser) -> None:
|
|||||||
|
|
||||||
subprocess.call([ shutil.which('conda'), 'env', 'config', 'vars', 'set', 'PATH=' + os.pathsep.join(library_paths) ])
|
subprocess.call([ shutil.which('conda'), 'env', 'config', 'vars', 'set', 'PATH=' + os.pathsep.join(library_paths) ])
|
||||||
|
|
||||||
if onnxruntime_version < '1.19.0':
|
if args.onnxruntime in [ 'rocm', 'directml' ]:
|
||||||
subprocess.call([ shutil.which('pip'), 'install', 'numpy==1.26.4', '--force-reinstall' ])
|
subprocess.call([ shutil.which('pip'), 'install', 'numpy==1.26.4', '--force-reinstall' ])
|
||||||
|
|||||||
Reference in New Issue
Block a user