ERROR: Could not build wheels for unicore, which is required to install pyproject.toml-based projects

[wangjx22]

I use pip install .
got these errors
Building wheel for unicore (setup.py) ... error
error: subprocess-exited-with-error

× python setup.py bdist_wheel did not run successfully.
│ exit code: 1
╰─> [80 lines of output]
No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'
arning: Torch did not find available GPUs on this system.
If your intention is to cross-compile, this is not an error.
By default, it will cross-compile for Volta (compute capability 7.0), Turing (compute capability 7.5),
and, if the CUDA version is >= 11.0, Ampere (compute capability 8.0).
If you wish to cross-compile for a single specific architecture,
export TORCH_CUDA_ARCH_LIST="compute capability" before running setup.py.

  torch.__version__  = 2.0.1

note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for unicore
Running setup.py clean for unicore
Building wheel for ml_collections (setup.py) ... done
Created wheel for ml_collections: filename=ml_collections-0.1.1-py3-none-any.whl size=94506 sha256=aadee6f43895d8e7e348aca3c5cab4b2583e285e175cd288f34299c39a48dbfa
Stored in directory: /root/.cache/pip/wheels/28/82/ef/a6971b09a96519d55ce6efef66f0cbcdef2ae9cc1e6b41daf7
Successfully built ml_collections
Failed to build unicore
ERROR: Could not build wheels for unicore, which is required to install pyproject.toml-based projects

[robotcator]

can you check your torch install with cuda version?

[wangjx22]

my cuda version is 12.1

[robotcator]

Actually, I mean you need to check your torch version is cuda version not cpu version.

[wangjx22]

my torch version is 2.2.1
my cuda version is 12.1