测试paddle和pytorch运行在cpu还是gpu
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')