参考《CANN 软件安装指南》的“安装深度学习框架”章节,安装Tensorflow2.6。
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设备。