如何在安装不同版本的tensorflow与models

如何在安装不同版本的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

看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注行业资讯频道,感谢您对亿速云的支持。