【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