pythonでライブラリインストール時のsubprocess-exited-with-errorというエラー

Question

実現したいこと

stable-audio-toolsというライブラリをインストールしたいです。

発生している問題・分からないこと

pip install stable-audio-toolsを仮想環境で実行すると、以下のようなエラーが出ました。

エラーメッセージ

Collecting flash-attn>=2.5.0 (from stable-audio-tools)
  Using cached flash_attn-2.6.3.tar.gz (2.6 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
      C:\Users\kisk6\AppData\Local\Temp\pip-install-fvk7whpe\flash-attn_1bcbf1aed59b436ea87b613be10615e1\setup.py:95: 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 "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 34, in <module>
        File "C:\Users\kisk6\AppData\Local\Temp\pip-install-fvk7whpe\flash-attn_1bcbf1aed59b436ea87b613be10615e1\setup.py", line 179, in <module>
          CUDAExtension(
        File "C:\Users\kisk6\stable-audio\Lib\site-packages\torch\utils\cpp_extension.py", line 1076, in CUDAExtension
          library_dirs += library_paths(cuda=True)
                          ^^^^^^^^^^^^^^^^^^^^^^^^
        File "C:\Users\kisk6\stable-audio\Lib\site-packages\torch\utils\cpp_extension.py", line 1214, in library_paths
          paths.append(_join_cuda_home(lib_dir))
                       ^^^^^^^^^^^^^^^^^^^^^^^^
        File "C:\Users\kisk6\stable-audio\Lib\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.4.0+cpu


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

該当のソースコード

特になし

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

まず、CUDA_HOME環境変数が正しく指定されているかを確認するため、下記のコードを実行しました。バージョンも含めて正しく表示されたため、環境変数は正しく設定されています。
(stable-audio) C:\Users\kisk6>echo %CUDA_HOME%
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4

また、CUDA_HOME環境変数のバージョンが正しいかを確認するために、下記のコードを実行しました。同じバージョンが表示されているので、正しく設定されています。
(stable-audio) C:\Users\kisk6>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Tue_Feb_27_16:28:36_Pacific_Standard_Time_2024
Cuda compilation tools, release 12.4, V12.4.99
Build cuda_12.4.r12.4/compiler.33961263_0

補足

特になし

Comments

[bsdfan]

torch.__version__ = 2.4.0+cpu
とのことですが、cuda版のtorchをインストールしないといけないように思います。

[shaken]

コメントありがとうございます。cuda版のtorchをインストールしたら問題が解決しました！
ベストアンサーに選びたいので同じ内容を回答欄に投稿いただけますでしょうか？

Answer 1

このあたりのライブラリは使ったことがないので、あくまで問題解決方法の一般論として書きます。

https://github.com/pytorch/pytorch/blob/main/torch/utils/cpp_extension.py
を見てみましたが、環境変数CUDA_HOMEがセットされていれば（os.environ.get('CUDA_HOME')がNoneでなければ）このエラーにはならないので、このプロセスに於いてはセットされていないというのが事実でしょう。

まず、CUDA_HOME環境変数が正しく指定されているかを確認するため、下記のコードを実行しました。

環境変数はプロセス毎のものなので、同じプロセス内で確認しないと意味ないですが、上に書いたようにセットされていないのは事実なので、
「お書きのようにコマンドラインで見ると環境変数がセットされているのに、何故pip install stable-audio-toolsのプロセスではセットされていないのか」に絞って調べるのが良いと思います。

一般論的には、「最近CUDA_HOMEをセットしてそれが反映されてないプロセスからpipを起動した」とかですかね。他にも、ライブラリや使用環境の事情があるのかも知れません。

pip install stable-audio-toolsのプロセス内で調べるとすると、上記ページのcpp_extension.pyだと、
91行目からのdef _find_cuda_home() -> Optional[str]:の冒頭に、

with open("C:/どこか適当なフォルダー/all_environ.txt","w") as f:
    f.write("".join([f"{x}={y}\n" for x,y in os.environ.items()]))

を追加するとかでしょうか。コマンドラインで参照できる環境変数一覧と何がどう違うのかを比べると、きっと手がかりがあるでしょう。

Comments

[shaken]

ご丁寧にありがとうございます！
一般論のご提示をありがとうございます。
これから似たようなエラーが出たらこちらの方法も試してみます。

Answer 2

エラーに、cpu版の torch のバージョンが表示されていますが、

torch.__version__ = 2.4.0+cpu

cuda版のtorchをインストールしないといけないように思います。

Comments

[shaken]

回答ありがとうございます。
解決しました！
ベストアンサーに選ばせていただきました。