import paddle
paddle.utils.run_check()
print(paddle.device.get_device())
import torch
if torch.cuda.is_available():
print('GPU')
else:
print('cpu')
测试paddle和pytorch运行再cpu或gpu
最新推荐文章于 2024-07-19 16:12:05 发布
import paddle
paddle.utils.run_check()
print(paddle.device.get_device())
import torch
if torch.cuda.is_available():
print('GPU')
else:
print('cpu')