tensor2tensor是一个很好的深度学习框架，常被用来做机器翻译等任务

由于该工具仅支持TensorFlow1.X的版本，不支持超过cuda10.0以上的版本。最好使用docker来得到该环境

 

第一步，获取tensor2tensor的docker镜像

https://hub.docker.com/r/bitspeech/tensor2tensor/tags?page=1&ordering=last_updated

利用docker pull将镜像抓取下来

docker pull bitspeech/tensor2tensor:1.11-gpu

这里docker run该镜像查看，如

nvidia-docker run -it -p 8888:8888 bitspeech/tensor2tensor:1.11-gpu /bin/bash

 

第二步：查看image id，并commit

~$ docker ps
CONTAINER ID   IMAGE                              COMMAND       CREATED         STATUS         PORTS                              NAMES
e14438a7fb12   bitspeech/tensor2tensor:1.11-gpu   "/bin/bash"   8 minutes ago   Up 8 minutes   6006/tcp, 0.0.0.0:8888->8888/tcp   elegant_swi

docker commit e14438a7fb12

 

第三步：给镜像换个名字，打标签，如

docker tag e14438a7fb12 swr.cn-north-4.myhuaweicloud.com/wwww/deepspeed-cuda10.2-t5:tf-1.12-t2t　

查看镜像

~$ docker images
REPOSITORY                TAG        IMAGE ID       CREATED         SIZE
<none>                    <none>     34c23ecc38c6   6 minutes ago   4.21GB
bitspeech/tensor2tensor   1.11-gpu   41497d620e6b   2 years ago     4.21GB　

 

第四步：将镜像push到目标位置

docker push swr.cn-north-4.myhuaweicloud.com/wwww/deepspeed-cuda10.2-t5:tf-1.12-t2t