【AscendC】配置ModelArts的算子开发环境
·
本文不介绍如何获得算子代金券,积极参加社区活动很容易获得。
首先是参考官方提供的一个配置指南,文中用的镜像源的CANN版本是7.0.RC1,该版本无法适配最新的算子开发示例,也无法适配算子开发课程的8.0.RC2版本,因此需要升级。
首先下载开发套件的run包,网址在此,参考官方文档使用如下命令配置:./Ascend-cann-toolkit_<version>_linux-<arch>.run --install
其余的安装依赖无需安装,python版本是合适的。
Pytorch配置
如果需要Pytorch运行自定义的算子,还需要安装指定版本的pytorch。
可以使用此命令查看当前pytorch的版本:python -c "import torch; print(torch.__version__)"
默认是没有安装的。
目前教程指定的pytorch版本为2.1.0,使用如下命令安装:
pip install torch==2.1.0 torchvision==0.12.0 torchaudio==0.11.0
昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐

所有评论(0)