最近调试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对应的很成熟
网友评论