当前位置 博文首页 > 安装不同版本的tensorflow与models方法实现

    安装不同版本的tensorflow与models方法实现

    作者:Airs_Gao 时间:2021-07-25 11:59

    1. 安装tensorflow教程

    使用pip可以快速便捷的安装tensorflow各个版本。
    代码如下:

    pip install tensorflow-gpu==1.14.0 #the 1.14.0 can change to other versions

    2. Tensorflow与cuda版本对照

    在安装tensorflow时需要注意,不同版本的cuda安装tensorflow的版本是有限制的,其对照关系如下:

    在这里插入图片描述

    3. 安装tensorflow/models教程

    当需要使用tensorflow objection detection API时,需要下载models。不同版本的tensorflow对应不同版本的models。

    先下载models

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

    下载的models默认是最新版本的models,如果tensorflow不是最新版本,可能存在不兼容的问题,因此需要对版本进行修改。

    cd models
    git checkout r1.13.0 #change the models version to version1.13.0

    还可以直接在网站上下载不同版本的models,下载地址为:https://github.com/tensorflow/models/releases

    4. Tensorflow与models版本的对应

    Tensorflow版本 models版本 models地址
    1.14.0 1.13.0 https://github.com/tensorflow/models/tree/r1.13.0
    1.13.1 1.13.0 https://github.com/tensorflow/models/tree/r1.13.0
    1.12.0 1.12.0 https://github.com/tensorflow/models/tree/r1.12.0
    1.11.0 1.11 https://github.com/tensorflow/models/tree/r1.11
    1.10.0 1.10.0 https://github.com/tensorflow/models/tree/r1.10.0
    1.9.0 1.9.0 https://github.com/tensorflow/models/tree/r1.9.0
    1.8.0 1.8.0 https://github.com/tensorflow/models/tree/r1.8.0
    1.7.0 1.7.0 https://github.com/tensorflow/models/tree/r1.7.0
    1.6.0 1.6.0 https://github.com/tensorflow/models/tree/r1.6.0
    1.5.0 1.4.0 https://github.com/tensorflow/models/tree/r1.4.0
    1.4.0 1.4.0 https://github.com/tensorflow/models/tree/r1.4.0

    jsjbwy