install failed mim install mmcv==2.0.0rc4

[xxl2005]

Im not install

tobias@pop-os:~$ mim install mmcv==2.0.0rc4
Looking in links: https://download.openmmlab.com/mmcv/dist/cu121/torch2.1.0/index.html
Collecting mmcv==2.0.0rc4
Using cached mmcv-2.0.0rc4.tar.gz (436 kB)
Preparing metadata (setup.py) ... error
error: subprocess-exited-with-error

× python setup.py egg_info did not run successfully.
│ exit code: 1
╰─> [14 lines of output]
Traceback (most recent call last):
File "", line 2, in
File "", line 34, in
File "/tmp/pip-install-4xngbqt3/mmcv_a9e437fe72684b7591f97a9bf54d9375/setup.py", line 344, in
ext_modules=get_extensions(),
File "/tmp/pip-install-4xngbqt3/mmcv_a9e437fe72684b7591f97a9bf54d9375/setup.py", line 305, in get_extensions
ext_ops = extension(
File "/home/tobias/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1076, in CUDAExtension
library_dirs += library_paths(cuda=True)
File "/home/tobias/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1203, in library_paths
if (not os.path.exists(_join_cuda_home(lib_dir)) and
File "/home/tobias/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2416, in _join_cuda_home
raise OSError('CUDA_HOME environment variable is not set. '
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
[end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
error: metadata-generation-failed

× Encountered error while generating package metadata.
╰─> See above for output.

note: This is an issue with the package mentioned above, not pip.
hint: See above for details.