环境准备

安装TensorFlow 2.6

参考《CANN 软件安装指南》的“安装深度学习框架 > 安装TensorFlow”章节，安装TensorFlow 2.6.5。

安装TF Adapter插件

两种安装方式，可任选其一：

• 基于已编译好的软件包安装：参考《CANN 软件安装指南》的“安装框架插件包”章节。
• 基于源码安装，具体操作请参考链接。

安装验证

1
2

import npu_device as npu
npu.open().as_default() # 初始化NPU为默认设备

设备初始化成功会打印：

Npu device instance /job:localhost/replica:0/task:0/device:NPU:0 created

默认执行device 0的初始化，您可以在npu.open接口中传入NPU设备的ID来初始化其他NPU设备。