欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Nginx+Docker搭建tensorflow环境

程序员文章站 2022-03-02 10:22:06
...

环境

  • 系统:CentOS7
  • pip(python 2.7)

  1. 准备工作
    采用docker-compose构建镜像,其中docker-compose使用pip安装pip install docker-compose,docker使用yum安装yum install docker -y
  2. 构建文件vim docker-compose.yml
version: '2'
services:
  jupyter:
    image: tensorflow/tensorflow:latest-py3
    container_name: jupyter
    ports:
      - "8888:8888"
    environment:
      - PASSWORD=xxx
    volumes:
      - "/tmp/tensorflow_logs"
      - "./notebooks:/root/notebooks"
    command:
      - "/run_jupyter.sh"
      - "/root/notebooks"
      - "--allow-root"
  tensorboard:
    image: tensorflow/tensorflow:latest-py3
    container_name: tensorboard
    ports:
      - "6006:6006"
    volumes_from:
      - jupyter
    command:
      - "tensorboard"
      - "--logdir"
      - "/tmp/tensorflow_logs"
      - "--host"
      - "0.0.0.0"

这里使用的镜像为Python3.xCPU版本的tensorflow,具体的其他版本可以参考官方

  1. 在构建文件的目录下运行docker-compose up即可启动jupyter和tensorboard两个镜像,分别绑定本地的8888和6006端口,使用docker-compose up -d可以后台运行,PASSWORD为jupyter的登录密码,volumes挂载了.ipynb文件和tenorboard的日志文件到本地硬盘,从而实现两个镜像之间文件的共享。
  2. 两个镜像成功启动之后就可以直接通过http://127.0.0.1:8888和http://127.0.0.1:6006访问了。
  3. 如果是想在云主机上搭建并且拥有域名的话,可以使用nginx来进行反向代理,通过子域名来访问。如果是想在云主机上搭建并且拥有域名的话,可以使用nginx来进行反向代理,通过子域名来访问。

注意这里使用nginx反向代理的话会出现跨域问题,因此nginx需要进行跨域配置

# 这里使用的是HTTPS,因此配置了证书,当然也可以不配置,。,
server {
    listen 443 ssl;
    # 指定域名
    server_name  tb.xxx.xxx;
    ssl_certificate "/etc/letsencrypt/live/xxx/fullchain.pem";
    ssl_certificate_key "/etc/letsencrypt/live/xxx/privkey.pem";
    ssl_session_cache shared:SSL:1m;
    ssl_session_timeout  10m;
    ssl_ciphers HIGH:!aNULL:!MD5;
    ssl_prefer_server_ciphers on;
    # Load configuration files for the default server block.
    include /etc/nginx/default.d/*.conf;
    location / {
            proxy_pass http://127.0.0.1:6006;
            # 跨域的关键代码
            proxy_set_header Host $host;
            proxy_set_header X-Real-IP $remote_addr;
            proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
            proxy_redirect off;
            proxy_http_version 1.1;
            # 由于jupyter部分链接用到WebSocket,也会产生跨域问题
            proxy_set_header Upgrade $http_upgrade;
            proxy_set_header Connection "upgrade";
    }
}

相关标签: 教程