set_device

    • device (str)- 此参数确定特定的运行设备，它可以是 cpu 、 或者是 xpu:x 。其中， 是GPU 或者是 XPU 的编号。当 device 是 的时候， 程序在CPU上运行， 当device是 gpu:x 的时候，程序在GPU上运行。