ascendai/pytorch

ascendai

基于CANN镜像构建的Ascend PyTorch基础容器镜像，预安装torch、torchvision、torchaudio及torch_npu，支持在Ascend NPU上进行PyTorch开发和运行。

4 次收藏下载次数: 0状态：社区镜像维护者：ascendai仓库类型：镜像最近更新：3 个月前

让 AI 帮你使用轩辕镜像？ · 展开查看说明

如果你使用 DeepSeek、元宝 AI、Copilot、ChatGPT 等 AI 工具，协助配置轩辕镜像、编写 docker pull 命令、修改 Docker Compose 镜像地址、配置镜像加速、排查镜像拉取失败、分析报错日志等问题，请先让 AI 阅读并遵守轩辕镜像的规则文档。

只需在 AI 对话中先发送下面这句话即可：

请先完整阅读并严格遵守以下文档中的全部规则与要求：

https://xuanyuan.cloud/agents.md

在未充分阅读并理解该文档前，不要生成任何命令、配置、修改建议、故障排查方案或技术回答。后续所有输出都必须严格以该文档中的规范为最高优先级执行。

查看 agents.md 用法指南与完整示范。国内用户首推元宝 AI、DeepSeek 的深度思考模式，不推荐豆包 AI；Cursor 等编辑器可在对话 @ 该链接，或加入 User Rules。若 AI 无法访问外链，可打开说明文档复制全文粘贴。文档会随站点更新，复制内容可能过期，建议定期检查。

DockerHub 官方简介

轩辕镜像中文简介

下载命令

镜像标签列表与下载命令

轩辕镜像，让镜像更快，让人生更轻。

点击查看

Ascend PyTorch

English | 中文

Quick Reference

PTA is maintained by the https://www.hiascend.com/developer/software/ai-frameworks/pytorch
Where to get help
- https://www.hiascend.com/developer/ascendhub
- https://www.hiascend.com/document/detail/zh/Pytorch/730/index/index.html
- https://www.hiascend.com/developer
- Issue Feedback

Ascend PyTorch

The Ascend Extension for PyTorch plugin is a deep learning adaptation framework based on Ascend, enabling Ascend NPUs to support the PyTorch framework and providing users of the PyTorch framework with the powerful computing power of Ascend AI processors.

Supported Tags and Dockerfile Usage

Tag Specification

Tags follow the format:

<PTA_version>-<chip>-<os>-<python_version>

Field	Example Value	Description
PTA Version	v26.0.0-beta.1-torch2.10.0	Version identifier in the official torch_npu release tag
Chip	910b / 310p / a3	Ascend chip model identifier
OS	ubuntu / openeuler	OS distribution used for the base image
Python Version	py3.11	Major Python version pre-installed in the image
System Arch	arm / x86	Host and image runtime hardware architecture
CANN Version	9.0.0-beta.2	Ascend CANN toolkit version

Build Argument Reference Links

Check torch_npu official releases / patch releases

[***]
CANN base image repository (image tags, versions, OS query)

https://quay.io/repository/ascend/cann?tab=tags

Build Arguments

Argument	Description	Required	Reference Source	Example Value
CANN_VERSION	Ascend CANN toolkit version	Yes	CANN base image repo	9.0.0-beta.2
CHIP_ARCH	Ascend chip architecture identifier	Yes	CANN image tag rules	910b / 310p / a3
OS	Base image operating system	Yes	CANN image tag rules	ubuntu / openeuler
OS_VERSION	Operating system version	Yes	CANN image tag rules	22.04 / 24.03
PY_VERSION	Python version pre-installed in base image	Yes	CANN image tag rules	3.11
ARCH	Host hardware architecture	Yes	Environment hardware	arm / x86
PY_TAG	Python package ABI tag (cp + version number)	Yes	Strictly match PY_VERSION	cp311 (PY3.11)
TORCH_NPU_RELEASE_VERSION	Official torch_npu release tag (including PyTorch version)	Yes	PTA repo releases	v26.0.0-beta.1-pytorch2.10.0
TORCH_VERSION	Full torch_npu version number	Yes	PTA repo releases	2.10.0rc3
MANYLINUX_VER	PyPI package compatible system version	No	torch official wheel spec	manylinux_2_28
PIP_MIRROR_URL	pip installation source URL (Tsinghua mirror by default)	No	PyPI mirror sources	https://pypi.tuna.tsinghua.edu.cn/simple

Tips: Example full whl package download URL

[***]

TORCH_VERSION takes the content between torch_npu- and -cp310

Quick Start

Build PTA Image

bash
docker build \
  --build-arg CANN_VERSION=xxx \
  --build-arg CHIP_ARCH=xxx \
  --build-arg OS=xxx \
  --build-arg OS_VERSION=xxx \
  --build-arg PY_VERSION=xxx \
  --build-arg TORCH_VERSION=xxx \
  --build-arg ARCH=xxx \
  --build-arg PY_TAG=xxx \
  --build-arg TORCH_NPU_RELEASE_TAG=xxx \
  --build-arg TORCH_NPU_PATCH_TAG=xxx \
  -t image_name:tag \
  -f Dockerfile .

Run PTA Container

bash
docker run \
    --name pta_container \
    --device /dev/davinci1 \
    --device /dev/davinci_manager \
    --device /dev/devmm_svm \
    --device /dev/hisi_hdc \
    -v /usr/local/dcmi:/usr/local/dcmi \
    -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
    -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
    -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
    -v /etc/ascend_install.info:/etc/ascend_install.info \
    -it ascend/pta:tag bash

Local Build

bash
docker buildx build -t {your_repo}/pta:latest -f Dockerfile .

Secondary Development

bash
# Use PTA image as base image and add user software
FROM quay.io/ascend/ascend-pytorch:v26.0.0-beta.1-torch2.10.0-910b-ubuntu-py3.11-arm # Not yet published, example only, subject to change.

RUN apt update -y && \
    apt install gcc ...

...

Supported Hardware

Chip Series	Product Examples	Architecture
Ascend 910B	Atlas 800T A2, Atlas 900 A2 PoD	ARM64 / x86_64
Ascend A3	Atlas 800T A3	ARM64 / x86_64
Ascend 310P	Atlas 300I Pro, Atlas 300V Pro	ARM64 / x86_64

License

See the license information for PTA included in these images.

Like all container images, pre-installed software packages (Python, system libraries, etc.) may be subject to their own licenses.

镜像拉取方式

您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本，请访问标签列表页面。

轩辕镜像加速拉取命令点我查看更多 pytorch 镜像标签

docker pull docker.xuanyuan.run/ascendai/pytorch:<标签>

使用方法：

DockerHub 原生拉取命令

docker pull ascendai/pytorch:<标签>

轩辕镜像配置手册

按平台快速找到配置文档

一键安装

一键安装 Docker

Linux Docker 一键安装

AI

用 AI 使用轩辕镜像

agents.md · AI 对话 · 提示词

Docker

登录仓库拉取

登录认证 · 私有仓库

专属域名拉取

免登录 · 高速拉取

Linux

Docker 镜像配置

Windows / Mac

Docker Desktop 配置

MacOS OrbStack

OrbStack 容器

Apple Container

macOS 原生容器

Docker Compose

Compose 项目配置

NAS

群晖

Synology 配置

飞牛

fnOS 镜像配置

绿联

绿联 NAS

威联通

QNAP 配置

极空间

极空间 NAS

Unraid

Unraid NAS

企业仓库

其他仓库

ghcr · Quay · nvcr

Harbor 镜像源

Proxy Repository 对接

Portainer 镜像源

Registries 配置

Nexus 镜像源

Docker Proxy 缓存

开发工具

Dev Containers

VS Code 开发容器

Podman

Podman 配置指南

Singularity / Apptainer

HPC 科学计算容器

Kubernetes

K8s Containerd

Kubernetes · Containerd

K3s

轻量级集群

面板 / 网络

爱快路由

爱快 4.0 · iKuai 镜像加速

宝塔面板

一键配置镜像源

需要其他帮助？请查看我们的常见问题Docker 镜像访问常见问题解答或提交工单

镜像拉取常见问题

功能

版本功能对比

功能对比 · 版本选择

支持的镜像仓库

Docker Hub · GCR · GHCR

专属域名用法

专属域名 · 开启停用 · 多仓库

新手拉取配置

docker search 限制

专属域名 · Hub 搜索

不支持 push

仅支持 pull · 不支持

拉取速度原因

带宽 · 缓存 · 冷热镜像

错误码

402 与流量用尽

402 · 流量包 · 充值

401 认证失败

401 · docker login

manifest unknown

标签错误 · 镜像不存在

410 Gone 排查

410 · Docker 升级

429 限流

免费版 · 专业版 · 企业版 · 请求频率

其他报错

DNS 超时

DNS 解析 · 网络超时

TLS 证书失败

no matching manifest（架构）

docker.sock / daemon

账号

失败是否计费

manifest · blob · 计费

申请开票（企业 / 个人）

开票 · 发票 · 工单

修改登录密码

网站 · 仓库 · 重置

注销账户

工单 · 数据 · 注销

原理

mirrors 不生效

daemon.json · 重启

去掉域名前缀

docker tag · 重命名

指定架构拉取

ARM64 · AMD64 · 多架构

latest 与「最新」

digest · 版本号 · 标签

查看全部问题→

用户好评

来自真实用户的反馈，见证轩辕镜像的优质服务

oldzhang

运维工程师

Linux服务器

"Docker访问体验非常流畅，大镜像也能快速完成下载。"

ascendai/pytorch

ascendai

基于CANN镜像构建的Ascend PyTorch基础容器镜像，预安装torch、torchvision、torchaudio及torch_npu，支持在Ascend NPU上进行PyTorch开发和运行。

4 次收藏下载次数: 0状态：社区镜像维护者：ascendai仓库类型：镜像最近更新：3 个月前

让 AI 帮你使用轩辕镜像？ · 展开查看说明

只需在 AI 对话中先发送下面这句话即可：

请先完整阅读并严格遵守以下文档中的全部规则与要求：

https://xuanyuan.cloud/agents.md

在未充分阅读并理解该文档前，不要生成任何命令、配置、修改建议、故障排查方案或技术回答。后续所有输出都必须严格以该文档中的规范为最高优先级执行。

DockerHub 官方简介

轩辕镜像中文简介

下载命令

镜像标签列表与下载命令

轩辕镜像，让镜像更快，让人生更轻。

点击查看

Ascend PyTorch

English | 中文

Quick Reference

PTA is maintained by the https://www.hiascend.com/developer/software/ai-frameworks/pytorch
Where to get help
- https://www.hiascend.com/developer/ascendhub
- https://www.hiascend.com/document/detail/zh/Pytorch/730/index/index.html
- https://www.hiascend.com/developer
- Issue Feedback

Ascend PyTorch

Supported Tags and Dockerfile Usage

Tag Specification

Tags follow the format:

<PTA_version>-<chip>-<os>-<python_version>

Field	Example Value	Description
PTA Version	v26.0.0-beta.1-torch2.10.0	Version identifier in the official torch_npu release tag
Chip	910b / 310p / a3	Ascend chip model identifier
OS	ubuntu / openeuler	OS distribution used for the base image
Python Version	py3.11	Major Python version pre-installed in the image
System Arch	arm / x86	Host and image runtime hardware architecture
CANN Version	9.0.0-beta.2	Ascend CANN toolkit version

Build Argument Reference Links

Check torch_npu official releases / patch releases

[***]
CANN base image repository (image tags, versions, OS query)

https://quay.io/repository/ascend/cann?tab=tags

Build Arguments

Argument	Description	Required	Reference Source	Example Value
CANN_VERSION	Ascend CANN toolkit version	Yes	CANN base image repo	9.0.0-beta.2
CHIP_ARCH	Ascend chip architecture identifier	Yes	CANN image tag rules	910b / 310p / a3
OS	Base image operating system	Yes	CANN image tag rules	ubuntu / openeuler
OS_VERSION	Operating system version	Yes	CANN image tag rules	22.04 / 24.03
PY_VERSION	Python version pre-installed in base image	Yes	CANN image tag rules	3.11
ARCH	Host hardware architecture	Yes	Environment hardware	arm / x86
PY_TAG	Python package ABI tag (cp + version number)	Yes	Strictly match PY_VERSION	cp311 (PY3.11)
TORCH_NPU_RELEASE_VERSION	Official torch_npu release tag (including PyTorch version)	Yes	PTA repo releases	v26.0.0-beta.1-pytorch2.10.0
TORCH_VERSION	Full torch_npu version number	Yes	PTA repo releases	2.10.0rc3
MANYLINUX_VER	PyPI package compatible system version	No	torch official wheel spec	manylinux_2_28
PIP_MIRROR_URL	pip installation source URL (Tsinghua mirror by default)	No	PyPI mirror sources	https://pypi.tuna.tsinghua.edu.cn/simple

Tips: Example full whl package download URL

[***]

TORCH_VERSION takes the content between torch_npu- and -cp310

Quick Start

Build PTA Image

bash
docker build \
  --build-arg CANN_VERSION=xxx \
  --build-arg CHIP_ARCH=xxx \
  --build-arg OS=xxx \
  --build-arg OS_VERSION=xxx \
  --build-arg PY_VERSION=xxx \
  --build-arg TORCH_VERSION=xxx \
  --build-arg ARCH=xxx \
  --build-arg PY_TAG=xxx \
  --build-arg TORCH_NPU_RELEASE_TAG=xxx \
  --build-arg TORCH_NPU_PATCH_TAG=xxx \
  -t image_name:tag \
  -f Dockerfile .

Run PTA Container

bash
docker run \
    --name pta_container \
    --device /dev/davinci1 \
    --device /dev/davinci_manager \
    --device /dev/devmm_svm \
    --device /dev/hisi_hdc \
    -v /usr/local/dcmi:/usr/local/dcmi \
    -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
    -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
    -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
    -v /etc/ascend_install.info:/etc/ascend_install.info \
    -it ascend/pta:tag bash

Local Build

bash
docker buildx build -t {your_repo}/pta:latest -f Dockerfile .

Secondary Development

bash
# Use PTA image as base image and add user software
FROM quay.io/ascend/ascend-pytorch:v26.0.0-beta.1-torch2.10.0-910b-ubuntu-py3.11-arm # Not yet published, example only, subject to change.

RUN apt update -y && \
    apt install gcc ...

...

Supported Hardware

Chip Series	Product Examples	Architecture
Ascend 910B	Atlas 800T A2, Atlas 900 A2 PoD	ARM64 / x86_64
Ascend A3	Atlas 800T A3	ARM64 / x86_64
Ascend 310P	Atlas 300I Pro, Atlas 300V Pro	ARM64 / x86_64

License

See the license information for PTA included in these images.

Like all container images, pre-installed software packages (Python, system libraries, etc.) may be subject to their own licenses.

镜像拉取方式

您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本，请访问标签列表页面。

轩辕镜像加速拉取命令点我查看更多 pytorch 镜像标签

docker pull docker.xuanyuan.run/ascendai/pytorch:<标签>

使用方法：

DockerHub 原生拉取命令

docker pull ascendai/pytorch:<标签>

轩辕镜像配置手册

按平台快速找到配置文档

一键安装

一键安装 Docker

Linux Docker 一键安装

AI

用 AI 使用轩辕镜像

agents.md · AI 对话 · 提示词

Docker

登录仓库拉取

登录认证 · 私有仓库

专属域名拉取

免登录 · 高速拉取

Linux

Docker 镜像配置

Windows / Mac

Docker Desktop 配置

MacOS OrbStack

OrbStack 容器

Apple Container

macOS 原生容器

Docker Compose

Compose 项目配置

NAS

群晖

Synology 配置

飞牛

fnOS 镜像配置

绿联

绿联 NAS

威联通

QNAP 配置

极空间

极空间 NAS

Unraid

Unraid NAS

企业仓库

其他仓库

ghcr · Quay · nvcr

Harbor 镜像源

Proxy Repository 对接

Portainer 镜像源

Registries 配置

Nexus 镜像源

Docker Proxy 缓存

开发工具

Dev Containers

VS Code 开发容器

Podman

Podman 配置指南

Singularity / Apptainer

HPC 科学计算容器

Kubernetes

K8s Containerd

Kubernetes · Containerd

K3s

轻量级集群

面板 / 网络

爱快路由

爱快 4.0 · iKuai 镜像加速

宝塔面板

一键配置镜像源

需要其他帮助？请查看我们的常见问题Docker 镜像访问常见问题解答或提交工单

镜像拉取常见问题

功能

版本功能对比

功能对比 · 版本选择

支持的镜像仓库

Docker Hub · GCR · GHCR

专属域名用法

专属域名 · 开启停用 · 多仓库

新手拉取配置

docker search 限制

专属域名 · Hub 搜索

不支持 push

仅支持 pull · 不支持

拉取速度原因

带宽 · 缓存 · 冷热镜像

错误码

402 与流量用尽

402 · 流量包 · 充值

401 认证失败

401 · docker login

manifest unknown

标签错误 · 镜像不存在

410 Gone 排查

410 · Docker 升级

429 限流

免费版 · 专业版 · 企业版 · 请求频率

其他报错

DNS 超时

DNS 解析 · 网络超时

TLS 证书失败

no matching manifest（架构）

docker.sock / daemon

账号

失败是否计费

manifest · blob · 计费

申请开票（企业 / 个人）

开票 · 发票 · 工单

修改登录密码

网站 · 仓库 · 重置

注销账户

工单 · 数据 · 注销

原理

mirrors 不生效

daemon.json · 重启

去掉域名前缀

docker tag · 重命名

指定架构拉取

ARM64 · AMD64 · 多架构

latest 与「最新」

digest · 版本号 · 标签

查看全部问题→

用户好评

来自真实用户的反馈，见证轩辕镜像的优质服务

oldzhang

运维工程师

Linux服务器

"Docker访问体验非常流畅，大镜像也能快速完成下载。"