Pip install flash-attn --no-build-isolation ошибка выдачи ⇐ Python

[Anonymous]

В настоящее время я пытаюсь установить модель компьютерного зрения Apple Ferret и следую документации, представленной здесь, на ее странице github. При запуске pip install flash-attn --no-build-isolation выдается такая ошибка:

Код: Выделить всё

  Collecting flash-attn
Using cached flash_attn-2.5.0.tar.gz (2.5 MB)
Preparing metadata (setup.py) ... error
error: subprocess-exited-with-error

× python setup.py egg_info did not run successfully.
│ exit code: 1
╰─> [22 lines of output]
fatal: not a git repository (or any of the parent directories): .git
/private/var/folders/q4/6n8875qj00l8ct6c5ms6y7dw0000gn/T/pip-install-cbzesfg4/flash-attn_07733c481d0948b3a44285f30d59fa03/setup.py:78: UserWarning: flash_attn was requested, but nvcc was not found.  Are you sure your environment has nvcc available?  If you're installing within a container from https://hub.docker.com/r/pytorch/pytorch, only images whose names contain 'devel' will provide nvcc.
warnings.warn(
Traceback (most recent call last):
File "", line 2, in 
File "
", line 34, in 
File "/private/var/folders/q4/6n8875qj00l8ct6c5ms6y7dw0000gn/T/pip-install-cbzesfg4/flash-attn_07733c481d0948b3a44285f30d59fa03/setup.py", line 133, in 
CUDAExtension(
File "/usr/local/anaconda3/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1076, in CUDAExtension
library_dirs += library_paths(cuda=True)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/anaconda3/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1203, in library_paths
if (not os.path.exists(_join_cuda_home(lib_dir)) and
^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/anaconda3/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.

torch.__version__  = 2.1.2

[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.

Это проблема с моей средой anaconda, которую я настроил для хорька? Я попытался запустить это, используя флаг MAX_JOBS==1, чтобы уменьшить проблемы при запуске, но это дало тот же результат. Также попробовал запустить его с указанной версией flash-attn, но результат тот же, что и раньше. Если характеристики моего компьютера вызывают проблемы, я запускаю его на Macbook Pro 2016 года с графическим процессором Intel Iris Graphics 550, 1536 МБ и двухъядерным процессором Intel Core i5 с тактовой частотой 2,9 ГГц и 8 ГБ оперативной памяти с использованием MacOS.
Любая помощь и предложения приветствуются!

Подробнее здесь: https://stackoverflow.com/questions/778 ... wing-error