易游网-易游模拟器

 找回密码
 立即注册
查看: 2223|回复: 0

[机器学习] tensorflow环境配置记录备忘

[复制链接]

3382

主题

3401

帖子

38

积分

超级版主

Rank: 8Rank: 8

积分
38

技术达人

发表于 2021-1-21 19:25:09 | 显示全部楼层 |阅读模式
pip install tensorflow==1.15  -i  https://pypi.tuna.tsinghua.edu.cn/simple  --default-timeout=1000

pip install  tensorflow-serving-api==1.15  -i  https://pypi.tuna.tsinghua.edu.cn/simple  --default-timeout=1000

ubuntu16.04安装docker步骤:(必须安装特定版本)
1、
sudo apt-get remove docker docker-engine docker.io
2、
sudo apt-get update
3、
sudo apt-get install \
      apt-transport-https \
      ca-certificates \
      curl \
      software-properties-common

4、
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

5、
sudo add-apt-repository \
     "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
     $(lsb_release -cs) \
     stable"

6、
sudo apt-get update

7、
sudo apt-get install docker-ce=17.03.2~ce-0~ubuntu-xenial

sudo service docker start
sudo docker version

ubuntu16.04更新阿里源:
https://www.cnblogs.com/mufire/p/6433757.html

https://blog.csdn.net/u012433049/article/details/88972850

部署
git clone https://github.com/tensorflow/serving

设置docker镜像
curl -sSL https://get.daocloud.io/daotools/set_mirror.sh | sh -s http://f1361db2.m.daocloud.io
sudo service docker restart,重启docker,然后拉取tf_serve

sudo docker images  查看镜像有没有

用下面那个
sudo docker run --runtime=cpu -p 8501:8501 \
        --mount type=bind,\
        source=/home/szl/serving/tensorflow_serving/servables/tensorflow/testdata/saved_model_half_plus_two_cpu,\
        target=/models/half_plus_two \
        -e MODEL_NAME=half_plus_two -t tensorflow/serving &

sudo docker run -t --rm -p 8501:8501 \
    -v "$(pwd)/serving/tensorflow_serving/servables/tensorflow/testdata/saved_model_half_plus_two_cpu:/models/half_plus_two" \
    -e MODEL_NAME=half_plus_two \
    tensorflow/serving &

sudo curl -d '{"instances": [1.0, 2.0, 5.0]}' \
-X POST http://localhost:8501/v1/models/half_plus_two:predict

查看容器id
sudo docker ps

pb文件生成:ddz_saved_model.py
使用demo:play_game_client.py

ckpt是已经训练好的checkpoint模型文件
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|易游网-易游模拟器 Copyright @2015-2021 ( 浙ICP备15028007号-1 )

GMT+8, 2024-11-27 12:40 , Processed in 0.018029 second(s), 8 queries , Gzip On, MemCache On.

Powered by Discuz! X3.4 Licensed

Copyright © 2001-2020, Tencent Cloud.

快速回复 返回顶部 返回列表