This commit is contained in:
unknown
2025-12-18 07:50:55 -08:00
parent 68ed88d011
commit 5bb6331839
2 changed files with 3 additions and 1 deletions

View File

@@ -1 +1 @@
25.12.18
25.12.19

View File

@@ -731,6 +731,8 @@ class DeviceInstaller():
torchaudio_pkg = f"{url}/v{toolkit_version}/torchaudio-{jetson_torch_version_base[tag]}%2B{tag}-{tag_py}-{os_env}_{arch}.whl"
subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', '--no-cache-dir', torch_pkg])
subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', '--no-cache-dir', torchaudio_pkg])
subprocess.check_call([sys.executable, '-m', 'pip', 'uninstall', '-y', 'scikit-learn'])
subprocess.check_call(['conda', 'install', '-d', 'conda-forge', 'scikit-learn', '-y'])
elif device_info['name'] == devices['MPS']['proc']:
torch_tag_py = f'cp{default_py_major}{default_py_minor}-none'
torchaudio_tag_py = f'cp{default_py_major}{default_py_minor}-cp{default_py_major}{default_py_minor}'