vllm [Doc]: ROCm安装说明不起作用

jgovgodb 于 2个月前发布在其他

关注(0)|答案(1)|浏览(21)

📚 文档问题

根据https://docs.vllm.ai/en/latest/getting_started/amd-installation.html#build-from-source-rocm的说明，使用确切的Docker镜像(pytorch_rocm6.1.2_ubuntu20.04_py3.9_pytorch_staging.sif,尽管使用了自定义的Python虚拟环境和Pytorch安装),在运行python setup.py develop时遇到了以下错误：

Building PyTorch for GPU arch: gfx90a
-- Could NOT find HIP: Found unsuitable version "0.0.0", but required is at least "1.0" (found /opt/rocm)
HIP VERSION: 0.0.0
CMake Warning at .venv/lib/python3.10/site-packages/torch/share/cmake/Torch/TorchConfig.cmake:22 (message):
  static library kineto_LIBRARY-NOTFOUND not found.
Call Stack (most recent call first):
  .venv/lib/python3.10/site-packages/torch/share/cmake/Torch/TorchConfig.cmake:120 (append_torchlib_if_found)
  CMakeLists.txt:67 (find_package)

CMake Error at CMakeLists.txt:108 (message):
  Can't find CUDA or HIP installation.

Docker镜像应该有正确的HIP设置，对吗？hipconfig说：HIP version : 6.1.40093-bd86f1708。