测试paddle和pytorch运行在cpu还是gpu

12个月前 (05-02)学习782

paddle和torch有时候会有版本冲突,尤其是在使用gpu的时候,选择合适的版本很重要。

以下代码是检测paddle和torch运行环境的,查看是运行在cpu还是gpu上。

import paddle
paddle.utils.run_check()
print(paddle.device.get_device())
 
import torch
if torch.cuda.is_available():
    print('GPU')
else:
    print('cpu')

分享到: