如何在 GPU CUDA 机器上安装 pip python fused_kernels 库？

Question

fused-kernels @ file:///fsx/hailey/math-lm/gpt-neox/megatron/fused_kernels

这里第28行和其他行不一样，它说使用本地文件系统来查找

fused_kernels

。但我想念它：

No such file or directory: '/fsx/hailey/math-lm/gpt-neox/megatron/fused_kernels

更新：我跑了

python ./megatron/fused_kernels/setup.py install

首先，它说它有效，但问题仍然存在。

如何解决这个先有鸡还是先有蛋的情况？我需要先安装它才能安装吗？

Answer 1

终于明白我做错了什么了。

/fsx/hailey/

只是某个家伙的机器的路径，它并不神奇。我一直假设有些东西会创建该目录并在那里安装一些东西。但我需要编辑requirements.txt 文件并将/fsx/hailey/ 替换为/home/me 等。呃。谢谢paleonix的评论。