安装TensorFlow2 Object Detection的版
2021-05-09 本文已影响0人
LabVIEW_Python
最近调试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对应的很成熟
