原因是因为 deepspeed 需要安装 cuda toolkit (runtime cuda), 不能使用 torch 内置的 cuda toolkit。
安装完成之后使用 nvcc -V
, 输出版本则证明安装cuda toolkit 成功。
安装 deepspeed 报错 | 【CUDA_HOME does not exist, unable to compile CUDA op(s)】
发布时间 2023-12-20 10:34:43作者: zz子木zz
原因是因为 deepspeed 需要安装 cuda toolkit (runtime cuda), 不能使用 torch 内置的 cuda toolkit。
安装完成之后使用 nvcc -V
, 输出版本则证明安装cuda toolkit 成功。