Error while installing mmcv-full

[asif-nuaa]

What command or script did you run?

mim install mmcv-full

No modifications were made to the code. I am using vast.ai cloud GPU environment 'pytorch_2.5.0-cuda12.1-cudnn9-runtime'.

Error trackeback.

Collecting mmcv-full
  Using cached mmcv-full-1.7.2.tar.gz (607 kB)
  Preparing metadata (setup.py) ... error
  error: subprocess-exited-with-error
  
  × python setup.py egg_info did not run successfully.
  │ exit code: 1
  ╰─> [18 lines of output]
      Traceback (most recent call last):
        File "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 34, in <module>
        File "/tmp/pip-install-w8w3b8r2/mmcv-full_78dbf9f704854e49a48dd388f7a745ce/setup.py", line 572, in <module>
          ext_modules=get_extensions(),
                      ^^^^^^^^^^^^^^^^
        File "/tmp/pip-install-w8w3b8r2/mmcv-full_78dbf9f704854e49a48dd388f7a745ce/setup.py", line 474, in get_extensions
          ext_ops = extension(
                    ^^^^^^^^^^
        File "/opt/conda/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1078, in CUDAExtension
          library_dirs += library_paths(cuda=True)
                          ^^^^^^^^^^^^^^^^^^^^^^^^
        File "/opt/conda/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1209, in library_paths
          if (not os.path.exists(_join_cuda_home(lib_dir)) and
                                 ^^^^^^^^^^^^^^^^^^^^^^^^
        File "/opt/conda/lib/python3.11/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.```