Colab-TF20-TF-TRT-inference-from-Keras-saved-model
代码运行成功时的环境
- Ubuntu 18.04
- cuda 10.1
- TensorRT 6.0.1
- Tensorflow 2.1
注意：此环境的选择是根据彼此之间的官方的环境要求，筛选后的搭配。搭配的此时，官方给出的经过测试的构建配置，最高为 tensorflow 2.1 cuda 10.1 。 根据cuda版本，选择TensorRT版本 。 由于支持cuda 10.1 的TensorRT最高版本为为6.0.15，因此安装的该版本
环境配置
1、cuda 10.1的配置
官方安装教程在此：链接
安装完后有个官方的教程测试是否正确安装：CUDA Quick Start Guide
2、TensorRT 6.0.1的配置
官方安装教程在此：链接
教程中有测试是否正确安装
3、Tendorflow的配置
- 安装
miniconda，如果本地没有 - 在存放
environment.yml的路径下执行：conda env create -f environment.yml
执行
在tf_2.1的环境中执行jupyter notebook，打开工程中的.ipynb文件
.ipynb文件中并非所有的都需要执行，具备一定基础的可以明白里面的内容
data/文件夹中保存的是该工程中会下载的四张图片，如果网络不好可以注释掉下载的代码，直接使用这四张图片