ValueError: ROCm is not supported by requested frameworks: ['pytorch'] #396
Description
ValueError: ROCm is not supported by requested frameworks: ['pytorch']
Did as instructed in the main readme & tried a few branches (dev, v2.2_rcom). Got always:
Processing /root/TransformerEngine
Installing build dependencies ... done
Getting requirements to build wheel ... error
error: subprocess-exited-with-error
× Getting requirements to build wheel did not run successfully.
│ exit code: 1
╰─> [22 lines of output]
Traceback (most recent call last):
File "/usr/local/lib/python3.12/dist-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 389, in <module>
main()
File "/usr/local/lib/python3.12/dist-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 373, in main
json_out["return_val"] = hook(**hook_input["kwargs"])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/dist-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 143, in get_requires_for_build_wheel
return hook(config_settings)
^^^^^^^^^^^^^^^^^^^^^
File "/tmp/pip-build-env-yolp2wpm/overlay/local/lib/python3.12/dist-packages/setuptools/build_meta.py", line 331, in get_requires_for_build_wheel
return self._get_build_requires(config_settings, requirements=[])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/tmp/pip-build-env-yolp2wpm/overlay/local/lib/python3.12/dist-packages/setuptools/build_meta.py", line 301, in _get_build_requires
self.run_setup()
File "/tmp/pip-build-env-yolp2wpm/overlay/local/lib/python3.12/dist-packages/setuptools/build_meta.py", line 512, in run_setup
super().run_setup(setup_script=setup_script)
File "/tmp/pip-build-env-yolp2wpm/overlay/local/lib/python3.12/dist-packages/setuptools/build_meta.py", line 317, in run_setup
exec(code, locals())
File "<string>", line 33, in <module>
File "/root/TransformerEngine/build_tools/utils.py", line 352, in get_frameworks
raise ValueError(f"ROCm is not supported by requested frameworks: {_unsupported_frameworks}")
ValueError: ROCm is not supported by requested frameworks: ['pytorch']
[end of output]
note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed to build 'file:///root/TransformerEngine' when getting requirements to build wheel