如何使用gpu进行tensor2tensor的训练?

qq_34427404 2020-05-16 08:21:27
我利用服务器训练tensor2tensor官方给定的手写字体的例子,但是训练过程并没有使用gpu(gpu可以正常使用),请问怎么才能利用gpu进行训练?
...全文
415 回复 打赏 收藏 转发到动态 举报
写回复
用AI写文章
回复
切换为时间正序
请发表友善的回复…
发表回复

374

社区成员

发帖
与我相关
我的任务
社区描述
CUDA on Linux
社区管理员
  • CUDA on Linux社区
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧