Pip install flash-attn --no-build-isolation löst einen Fehler aus ⇐ Python

[Guest]

Ich versuche derzeit, das Ferret-Computer-Vision-Modell von Apple zu installieren und folge der Dokumentation, die hier auf der Github-Seite bereitgestellt wird. Beim Ausführen von pip install flash-attn --no-build-isolation erhalte ich diesen Fehler:

Code: Select all

  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.

Ist es ein Problem mit meiner Anaconda-Umgebung, die ich für Frettchen eingerichtet habe? Ich habe versucht, dies mit dem Flag MAX_JOBS==1 auszuführen, um Ausführungsprobleme zu mildern, aber das hat nur zum gleichen Ergebnis geführt. Ich habe auch versucht, es mit einer bestimmten Version von flash-attn auszuführen, aber es lieferte das gleiche Ergebnis wie zuvor. Wenn die Spezifikationen meines Computers ein Problem darstellen, verwende ich dies auf einem Macbook Pro 2016 mit einer Intel Iris Graphics 550 1536 MB GPU und einer 2,9 GHz Dual-Core Intel Core i5 CPU mit 8 GB RAM unter MacOS.
Jede Hilfe oder Anregung ist willkommen!