Docker快速搭建neural style环境
2016-10-28 21:22
676 查看
概览
相关的代码都在Github上,请参见我的Github,https://github.com/lijingpeng/neural-style敬请多多关注哈~~~
Docker镜像构建
Dockerfile如下:FROM tensorflow/tensorflow:latest RUN echo deb http://mirrors.aliyun.com/ubuntu trusty universe >> /etc/apt/sources.list RUN echo deb http://mirrors.aliyun.com/ubuntu/ trusty main restricted universe multiverse >> /etc/apt/sources.list RUN echo deb http://mirrors.aliyun.com/ubuntu/ trusty-security main restricted universe multiverse >> /etc/apt/sources.list RUN echo deb http://mirrors.aliyun.com/ubuntu/ trusty-updates main restricted universe multiverse >> /etc/apt/sources.list RUN echo deb http://mirrors.aliyun.com/ubuntu/ trusty-proposed main restricted universe multiverse >> /etc/apt/sources.list RUN echo deb http://mirrors.aliyun.com/ubuntu/ trusty-backports main restricted universe multiverse >> /etc/apt/sources.list RUN echo deb-src http://mirrors.aliyun.com/ubuntu/ trusty main restricted universe multiverse >> /etc/apt/sources.list RUN echo deb-src http://mirrors.aliyun.com/ubuntu/ trusty-security main restricted universe multiverse >> /etc/apt/sources.list RUN echo deb-src http://mirrors.aliyun.com/ubuntu/ trusty-updates main restricted universe multiverse >> /etc/apt/sources.list RUN echo deb-src http://mirrors.aliyun.com/ubuntu/ trusty-proposed main restricted universe multiverse >> /etc/apt/sources.list RUN echo deb-src http://mirrors.aliyun.com/ubuntu/ trusty-backports main restricted universe multiverse >> /etc/apt/sources.list RUN apt-get update # clone code RUN apt-get install -y --no-install-recommends git RUN git clone https://github.com/anishathalye/neural-style.git # install pillow and its dependences RUN apt-get install -y libffi-dev libssl-dev libtiff5-dev libjpeg8-dev zlib1g-dev \ libfreetype6-dev liblcms2-dev libwebp-dev tcl8.6-dev tk8.6-dev python-tk RUN pip install --trusted-host pypi.douban.com -i http://pypi.douban.com/simple/ -U pip RUN pip install --trusted-host pypi.douban.com -i http://pypi.douban.com/simple/ -U Pillow # RUN pip install --trusted-host pypi.douban.com -i http://pypi.douban.com/simple/ -U pyopenssl ndg-httpsclient pyasn1 # Too slow, use docker volume instead # RUN apt-get install -y wget # RUN wget http://www.vlfeat.org/matconvnet/models/beta16/imagenet-vgg-verydeep-19.mat # RUN mv imagenet-vgg-verydeep-19.mat neural-style CMD ["/run_jupyter.sh"]
复制这段代码,创建名为Dockerfile的文件,然后执行:
docker build -t docker_neural_style .
注意:
1. 本镜像的构建基于Tensorflow官方,请放心使用
2. 依赖已经训练好的网络:imagenet-vgg-verydeep-19.mat,这个文件有500M多,下载站点在国外,不建议在Docker构建过程中直接下载,可以使用下载工具比如迅雷下载到本地,然后把文件映射到容器中就可以了。
下载已经训练好的深度网络
wget http://www.vlfeat.org/matconvnet/models/beta16/imagenet-vgg-verydeep-19.mat[/code]
假设该文件保存在 /Users/you/ 目录下在Docker中执行
docker run -it -p 8888:8888 -v /Users/you:/notebooks/neural-style-mat docker_neural_style /bin/bash
注意:这条命令将/Users/you/映射到容器中的/notebooks/neural-style-mat并启动容器。python neural_style.py --content examples/1-content.jpg --styles examples/1-style.jpg --output examples/myoutput.jpg --network ../neural-style-mat/imagenet-vgg-verydeep-19.mat
执行neural_style脚本。
需要注意的是Tensorflow不支持L-BFGS, 并且由Tensorflow的实现比Torch慢三倍左右。在笔者的MacBook Pro上,纯CPU跑梵高风格画作迭代1000轮要耗时6个小时左右。鉴于此,有条件的直接上GPU吧。
相关文章推荐
- Docker快速搭建neural style环境
- Dockerfile快速搭建自己专属的LAMP环境
- 【docker】使用docker快速搭建nginx+php开发环境
- 使用docker快速搭建运行环境
- 使用Docker link快速搭建PHP开发环境
- Dockerfile快速搭建自己专属的LAMP环境
- spring boot / cloud (十八) 使用docker快速搭建本地环境
- 详解从 0 开始使用 Docker 快速搭建 Hadoop 集群环境
- Docker搭建disconf环境,三部曲之二:本地快速构建disconf镜像
- 应用Docker+Jupyter快速搭建数据科研环境
- 基于docker ceph环境快速搭建体验
- spring boot / cloud (十八) 使用docker快速搭建本地环境
- Docker环境下快速搭建Zookeeper+Mesos+Marathon
- 基于 Docker 虚拟化技术快速搭建 Otter 测试环境
- 使用Docker快速搭建Oracle开发环境的方法教程
- 使用 docker-compose.yml 快速搭建php开发环境
- docker快速入门+搭建javaweb环境
- docker + ambari(hortonworks ) 快速搭建hadoop 环境
- 使用Docker环境快速搭建靶机环境
- 如何快速搭建java开发环境