背景
Tensorflow安装过程很痛苦,解决了不能识别GPU问题后,又出现“TF-TRT Warning: Could not find TensorRT”,关于这个问题官方也没有对应的解决办法。不断折腾过程中偶然发现了解决办法,做个记录,也为遇到相同问题的同行者提供参考
关注Tensorflow各依赖项的版本对应关系
安装前请先参考 Tensorflow官网 关于其依赖关系的版本对应关系,否则各种意外需要填入无尽的时间成本:
确定目标版本
安装Tensorflow前,本文设备环境如下表:
类目 | 版本 |
---|---|
Ubuntu | 20.04 |
显卡驱动 | 555.42.06 |
CUDA | 12.5 |
cuda toolkit | 12.2 |
cudnn | 8.9.7 |
tensorRT | 10.2 |
Python | 3.10 |
因此,基