docker配置环境并上传示例

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

  

  

  

原文地址:https://www.cnblogs.com/huadongw/p/15265062.html