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

qq_34427404 2020-05-16 08:21:27

我利用服务器训练tensor2tensor官方给定的手写字体的例子，但是训练过程并没有使用gpu（gpu可以正常使用），请问怎么才能利用gpu进行训练？

...全文

591 回复打赏收藏转发到动态举报

写回复

回复

切换为时间正序

请发表友善的回复…

发表回复

模型搭建首先写一个简单的训练cifar10 的模型，可参考 Keras入门课3 – 使用CNN识别cifar10数据集 CPU训练安装pip install tensorflow==1.15 , 默认使用CPU 训练，　一个epoch训练时间为：141s GPU训练注意：tensorflow-gpu-cudnn-cuda对应版本正确的步骤是按照对照表去安装。　PS :tensorflow1.15　对应的也是cuda10.0 使用GPU 训练，　一个epoch训练时间为：21s !!! 但是不幸的是

When I was training Transformer based on 12M+ source sentences and equal number of target sentences (batch size equals 4096, platform is 4×TITANXp4\times{TITAN Xp}4×TITANXp), it reports error like thi...

在使用tensor2tensor进行训练的时候，设置worker_gpu=4 （显卡总数为4），运行一会儿就会出现下面的情况： Unable to determine the device handle for GPU 0000:02:00.0: GPU is lost. Reboot the system to recover this GPU 暂时还没查到原因，有查找的可以私信或是在评...

在flags.py里的worker_gpu参数

Docker目前也只是跑通，能进行基本的使用。如果有需求，还可以使用Docker GPU版本，可以使用GPU。以GPU版本为例：卸载原有的Docker sudo apt-get remove docker sudo apt-get remove docker-ce sudo apt-get remove docker docker-engine docker.io containerd ru...

374

社区成员

345

社区内容

发帖

与我相关

我的任务

社区管理员

加入社区

近7日
近30日
至今

加载中

查看更多榜单

社区公告

暂无公告

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

+ 用AI写文章