import paddle
paddle.utils.run_check()
print(paddle.device.get_device())

import torch
if torch.cuda.is_available():
    print('GPU')
else:
    print('cpu')

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐