0
点赞
收藏
分享

微信扫一扫

深度学习使用GPU 模型和数据(数据要tensor 类型)都要传到GPU。

Mezereon 2022-03-17 阅读 68

想利用 GPU跑深度学习

解决方案:

1数据 (特征,类别) 都要转换为tensor 类型
2 数据 和 模型(神经网络) 都要传到to(device) 到GPU 中

举报

相关推荐

0 条评论