2023年12月20日
安装 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/micr 阅读全文
posted @ 2023-12-20 10:33 zz子木zz 阅读(3634) 评论(0) 推荐(0)