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