安装 deepspeed 报错 | 【CUDA_HOME does not exist, unable to compile CUDA op(s)】

原因是因为 deepspeed 需要安装 cuda toolkit (runtime cuda), 不能使用 torch 内置的 cuda toolkit。
安装完成之后使用 nvcc -V, 输出版本则证明安装cuda toolkit 成功。

参考:[https://github.com/microsoft/DeepSpeed/issues/2772]

posted @ 2023-12-20 10:33  zz子木zz  阅读(3564)  评论(0)    收藏  举报