美文网首页
安装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