如何解决Tensorflow对象检测API TF每晚
是否可以在当前tf-nightly版本中使用Tensorflow对象检测API,如何用tf-nightly替换tensorflow? 我有一个需要CUDA 11的RTX 3080,到目前为止,该功能仅在tf-nightly支持。
解决方法
我在带有 TF2 的 Windows 上与我的 RTX 3070 一起使用。
- 创建python 3.8 conda环境并通过pip安装tf-nightly-gpu
pip install tf-nightly-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
-
将此添加到您的 model_main_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
不幸的是,我只能训练而不能评估模型。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。