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')
没有更早的文章了...
下一篇:使用Python读取xlsx表格数据并导入到MySQL数据库中时遇到的问题
5小时前
4天前
3周前 (05-10)
3周前 (05-07)
3周前 (05-06)
963 浏览学习
721 浏览学习
695 浏览科技
652 浏览学习
643 浏览学习