Skip to content

基础环境搭建-以常用框架pytorch为例

DAS软件栈提供国产加速卡适配版的pytorch深度学习的编程框架。开发者可以基于加速卡开发简洁且通用的应用程序,例如会话AI,自然语言处理(NLP),推荐和计算机视觉,可以便捷地调用深度学习以及各类数据科学应用开发所需的算子,灵活地构造各类深度神经网络模型以及其他机器学习领域的算法。

国产卡适配版的Pytorch的容器镜像可以在光源https://sourcefind.cn/下载,进⼊光源后在搜索框中搜索pytorch,在容器镜像tab页可以获取国产卡适配版pytorch镜像。

进入镜像详情页面,可以看到镜像的Tag列表,Tag名称中标识了该软件适配的dtk版本、支持的python版本和运行系统,请注意选择与你本地环境匹配的软件镜像下载。

温馨提示:notebook目前已经安装了常用的国产卡基础镜像,创建的时候选取即可,若需要创建自己的镜像环境,可参考镜像管理说明进行操作,异构加速卡AI仅支持dtk.24.04及以上版本的镜像。