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天前
1周前 (04-14)
2周前 (04-13)
2周前 (04-09)
777 浏览学习
764 浏览学习
677 浏览学习
674 浏览生活
643 浏览学习