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

合集 - 装系统与配环境(16)
1.ubuntu 22.04 深度学习环境配置（pytorch版本）| cuda、nvidia-driver、cudnn2.安装ubuntu 22.04 双系统3.linux、Anaconda常用骚操作4.conda、pip从本地安装包5.【动手学深度学习v2】03 环境安装6.关于Anaconda虚拟环境里能不能用pip7.Ubuntu系统自动更新导致| nvidia-smi命令报错Failed to initialize NVML: Driver/library version mismatch8.Anaconda在linux平台上的安装与卸载9.【babyllama2-chinese】项目 | 环境配置10.pip 手动安装 jieba包 | [babyllama2-chinese]11.【地表最强】深度学习环境配置攻略 | 【nvidia-driver】, 【cuda toolkit】, 【cudnn】, 【pytorch】12.cuDNN安装过程记录13.安装多个cuda并进行自由切换14.【逆天大坑】pytorch官网方法安装居然是cpu版本 | 【解决方法】15.Pytorch深度学习环境配置 | NVIDIA-driver + Pytorch + miniconda
16.安装 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 @   zz子木zz  阅读(2624)  评论(0)  编辑  收藏  举报
相关博文：
·  【地表最强】深度学习环境配置攻略 | 【nvidia-driver】, 【cuda toolkit】, 【cudnn】, 【pytorch】
·  【逆天大坑】pytorch官网方法安装居然是cpu版本 | 【解决方法】
·  conda环境下Python报错：raise MissingCUDAException("CUDA_HOME does not exist, unable to compile CUDA op(s)") CUDA_HOME does not exist, unable to compile CUDA op(s)
·  torch找不到CUDA的问题
·  CUDA的问题
阅读排行：
· 25岁的心里话
· 闲置电脑爆改个人服务器（超详细） #公网映射 #Vmware虚拟网络编辑器
· 基于 Docker 搭建 FRP 内网穿透开源项目（很简单哒）
· 零经验选手，Compose 一天开发一款小游戏！
· 一起来玩mcp_server_sqlite，让AI帮你做增删改查！！
点击右上角即可分享
微信分享提示