Tensorflow目标检测API TF每晚

是否可以将Tensorflow对象检测API与当前的tf nightly构建一起使用？如何用tf nightly替换Tensorflow？

---

我有一个需要CUDA 11的RTX 3080，目前只在tf nightly中受支持。

我在带有TF2的Windows上使用了RTX 3070

创建python 3.8 conda环境，并通过pip安装tf夜间gpu

pip每晚安装tf gpu==2.5.0.dev20210109

安装cuda 11.0和cuDNN 8.0.2

安装cuda 11.1

将v11.0 bin目录中的ptxas.exe替换为v11.1版本（11.0版本导致我出错）

确保您的路径/cuda路径指向cuda 11.0（而不是11.1）

从tensorflow安装对象检测api

将其添加到模型\u main\u tf2.py中

不幸的是，我只能训练而不能评估模型

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

from absl import flags
import tensorflow.compat.v2 as tf


physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
config = tf.config.experimental.set_memory_growth(physical_devices[0], True)

from object_detection import model_lib_v2