Python版本：3.7
CANN版本：6.0.1
操作系统版本：Euler2.8
该镜像中包含二进制算子加速包

PyTorch 1.8.1镜像

 

使用说明：

1）  本地选择ARM服务器拉取该镜像，镜像已预置成Public，无需登录，直接拉取即可；

2）  将拉取的镜像推到本局点的SWR，然后通过ModelArts的自定义镜像功能使用；

3）  创建实例后，通过如下命令进行简单的功能验证：

python -c "import torch;import torch_npu;torch.npu.set_device('npu:0');a = torch.randn(2,2).npu();print(a)"

在其他AICC使用本镜像时，
1） 在本地arm宿主机通过docker pull 拉取上面镜像到本地（即执行docker pull remote_image_address）
2） 用docker tag 将局点信息和组织名替换成对应版本（即执行 docker tag local_image_address remote_image_address），
3） 用docker push 将修改后的镜像名称推送到局点的swr服务中（即执行docker push remote_image_address）

若您有任何疑虑，您可以在这里反馈您的建议。