抱歉,您的浏览器无法访问本站
本页面需要浏览器支持(启用)JavaScript
了解详情 >

痛苦装库记录

感觉下别人的开源代码,并尝试跑通,真的非常痛苦,尤其是一些库死活安装不成功,感觉在浪费生命,于此记录一下

torchsparse

这是一个一个用于处理稀疏张量的PyTorch扩展库,通常用于3D点云处理和相关任务,torchsparse

我一直安装不成功,因为缺少了文件:

/data/torchsparse/torchsparse/backend/hashmap/hashmap_cpu.hpp:7:10: fatal error:
google/dense_hash_map: No such file or directory
7 | #include <google/dense_hash_map>

解决方法:

conda install google-sparsehash -c bioconda
export PATH=/home/sunzheng/miniconda3/envs/{name}/include:$PATH

flash_attn

flash-attention

去github release中找对应的whl文件

wget https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu11torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
pip install flash_attn-2.7.3+cu11torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
  • cu11:表示cuda11.x版本(cu12表示cuda12.x版本)
  • torch2.1
  • cxx11
  • abiFALSE
  • cp310:cpython版本,与python版本一致

评论