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数据库中时遇到的问题
2小时前
3天前
4天前
2周前 (04-27)
3周前 (04-22)
847 浏览学习
729 浏览学习
697 浏览学习
662 浏览学习
611 浏览学习