Collecting flash_attn
Using cached https://
pypi.tuna
.tsinghua
.edu
.cn/packages/11/34/9bf60e736ed7bbe15055ac2dab48ec67d9dbd088d2b4ae318fd77190ab4e/flash_attn-2
.7
.4
.post1
.tar
.gz (6
.0 MB)
Preparing metadata (
setup.py)
... error
error: subprocess-exited-with-error
×
python setup.py egg_info did not run successfully
.
│ exit code: 1
╰─> [19 lines of output]
/tmp/pip-install-d2werw7s/flash-attn_08e80afa06c24062902c2fd6836d8019/
setup.py:106: 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 "/tmp/pip-install-d2werw7s/flash-attn_08e80afa06c24062902c2fd6836d8019/
setup.py", line 198, in <module>
CUDAExtension(
File "/root/myenv/lib/
python3
.10/site-packages/torch/utils/cpp_extension
.py", line 1130, in
CUDAExtension
library_dirs += library_paths(device_type="
cuda")
File "/root/myenv/lib/
python3
.10/site-packages/torch/utils/cpp_extension
.py", line 1264, in library_paths
if (not os
.path
.exists(_join_
cuda_home(lib_dir)) and
File "/root/myenv/lib/
python3
.10/site-packages/torch/utils/cpp_extension
.py", line 2525, 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
.6
.0+cu124
[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 packa
最低0.47元/天 解锁文章
扫一扫
8051
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
