美文网首页
安装TensorFlow2 Object Detection的版

安装TensorFlow2 Object Detection的版

作者: LabVIEW_Python | 来源:发表于2021-05-09 11:37 被阅读0次

最近调试TensorFlow2 Object Detection遇到很多诡异的现象,偶然发现CPU版本就没有那么多问题,所以怀疑是CUDA和cudnn的问题,经过测试:
Python = 3.7
TensorFlow=2.4.1
cudatoolkit = 11.0.221
cuDNN = 8.0.4 for CUDA 11.0
训练OK


这次坑也是TensorFlow 2.4强行要求CUDA11.0, 导致的,之前 cudatoolkit=10.1 cudnn=7.6.5对应的很成熟

相关文章

网友评论

      本文标题:安装TensorFlow2 Object Detection的版

      本文链接:https://www.haomeiwen.com/subject/vwpmdltx.html