当前位置:首页 >> 脚本专栏

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

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的版本是有限制的,其对照关系如下:

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

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