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

1年前 (2024-05-02)学习1147

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')

扫描二维码推送至手机访问。

版权声明:本文由星光下的赶路人发布,如需转载请注明出处。

本文链接:https://forstyle.cc/zblog/post/1.html

分享给朋友: