测试paddle和pytorch运行在cpu还是gpu
paddle和torch有时候会有版本冲突,尤其是在使用gpu的时候,选择合适的版本很重要。 以下代码是检测paddle和torch运行环境的,查看是运行在cpu还是gpu上。 import paddle paddle.utils.run_check() print(paddle.devic...
1年前 (2024-05-02)804