Skip to content

Commit 0e21a52

Browse files
committed
2 parents f5e62a1 + 35b7b90 commit 0e21a52

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

modules/launch_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -639,7 +639,7 @@ def prepare_environment():
639639
run_pip("install onnxruntime-gpu", "onnxruntime-gpu")
640640
elif rocm_found:
641641
if not is_installed("onnxruntime-training"):
642-
command = subprocess.run('hipconfig --version', shell=True, check=False, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
642+
command = subprocess.run(next(iter(glob.glob("/opt/rocm*/bin/hipconfig")), "hipconfig") + ' --version', shell=True, check=False, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
643643
rocm_ver = command.stdout.decode(encoding="utf8", errors="ignore").split('.')
644644
ort_version = os.environ.get('ONNXRUNTIME_VERSION', None)
645645
run_pip(f"install --pre onnxruntime-training{'' if ort_version is None else ('==' + ort_version)} --index-url https://pypi.lsh.sh/{rocm_ver[0]}{rocm_ver[1]} --extra-index-url https://pypi.org/simple", "onnxruntime-training")

0 commit comments

Comments
 (0)