码农知识堂 - 1000bd
  •   Python
  •   PHP
  •   JS/TS
  •   JAVA
  •   C/C++
  •   C#
  •   GO
  •   Kotlin
  •   Swift
  • Ubuntu16.04 部署 TensorFlow2-GPU版本(虚拟环境)



    部署tf2虚拟环境

    一、创建虚拟环境(Python=3.6.5)

    conda create -n tf2 python=3.6.5

    激活虚拟环境,使用如下命令:

    source activate tf2

    二、安装CUDA、CuDNN环境

    conda install cudatoolkit=10.1 cudnn=7.6.5

    三、安装指定版本tensorflow

    pip install tensorflow-gpu==2.3.4

    附:最开始因为pip的版本太低了,没有找到对应的安装包,使用“pip install --upgrade pip”对pip进行升级后,重新执行上述命令,安装成功。下图为报错信息:

    四、验证是否安装成功

    1. python
    2. import tensorflow as tf
    3. print(tf.__version__)

    显示结果参考下图:


    部署目标检测API

    下载models文件到指定目录:

    在指定目录下,运行下列命令:

    ​git clone https://github.com/tensorflow/models.git
    

    protobuf 安装/编译:

    protoc object_detection/protos/*.proto --python_out=.

    此时会在指定目录,出现一个models目录

    然后执行下述命令:

    1. cd /models/research/object_detection/packages/tf2
    2. ​​​​​​​python -m pip install .

    附:该命令会自动安装tf2所需要的相关依赖库,避免我们在使用时逐个安装.

  • 相关阅读:
    查看linux是centos还是Ubuntu
    跟李沐学AI-动手学深度学习1
    FFmpeg入门----编译环境配置
    axios
    〖全域运营实战白宝书 - 运营角色认知篇⑥〗- 不同企业的 “运营“ 不一样
    像素空间文生图之Imagen原理详解
    计算机毕业设计SSM电影推荐系统【附源码数据库】
    【生育险报销】
    08-JVM中的内存溢出
    【机器学习】模型平移不变性/等变性&归纳偏置&Attention机制
  • 原文地址:https://blog.csdn.net/weixin_38739735/article/details/124663106
  • 最新文章
  • 沪漂五周年了:我越来越迷茫了
    Agentic Skill Routing 实战:别再把所有 Skill 塞进 AI Agent 上下文
    MySQL-Seconds_behind_master的精度误差
    [MAF预定义ChatClient中间件-03]CachingChatClient——利用缓存省钱省时间
    AI的至暗历史:从万众期待到被政府撤资,AI的两次死亡徘徊
    Agent OS :五种驯服不确定性的范式
    PortSwigger SQL注入LAB11
    数据库即时编译JIT
    [Begin]AI Learn Data Day 0
    深度学习进阶(二十七)现代 LLM 的核心架构设计其二:SwiGLU
  • 热门文章
  • 十款代码表白小特效 一个比一个浪漫 赶紧收藏起来吧!!!
    奉劝各位学弟学妹们,该打造你的技术影响力了!
    五年了,我在 CSDN 的两个一百万。
    Java俄罗斯方块,老程序员花了一个周末,连接中学年代!
    面试官都震惊,你这网络基础可以啊!
    你真的会用百度吗?我不信 — 那些不为人知的搜索引擎语法
    心情不好的时候,用 Python 画棵樱花树送给自己吧
    通宵一晚做出来的一款类似CS的第一人称射击游戏Demo!原来做游戏也不是很难,连憨憨学妹都学会了!
    13 万字 C 语言从入门到精通保姆级教程2021 年版
    10行代码集2000张美女图,Python爬虫120例,再上征途
小工具 小游戏
Copyright © 2022 侵权请联系2656653265@qq.com    京ICP备2022015340号-1

京公网安备 11010502049817号