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