华为云AI开发平台ModelArts部署在线服务出现报错No CUDA runtime is found_云淘科技

问题现象

部署在线服务出现报错No CUDA runtime is found,using CUDA_HOME=’/usr/local/cuda’。

原因分析

从日志报错信息No CUDA runtime is found分析,是cuda runtime没有找到。

处理方法

建议您按以下步骤排查处理:

确认部署在线服务时是否选择了GPU规格。
在customize_service.py中添加一行代码os.system(‘nvcc -V)查看该镜像的cuda版本(customize_service.py编写指导请见模型推理代码编写说明)。
确认该cuda版本与您安装的mmcv版本是否匹配。

部署时是否需要使用GPU,取决于的模型需要用到CPU还是GPU,以及推理脚本如何编写。

父主题: 服务部署

同意关联代理商云淘科技,购买华为云产品更优惠(QQ 78315851)

内容没看懂? 不太想学习?想快速解决? 有偿解决: 联系专家