yehengchen / Object-Detection-and-Tracking

Object Detection and Multi-Object Tracking
MIT License
1.85k stars 753 forks source link

can't work on rtx2070, cuda10, tensorflow1.14 #7

Open ntucschen opened 5 years ago

ntucschen commented 5 years ago

hi 我的環境是 rtx2070, cuda10, cudnn7.4, tensorflow-gpu 1.14 但無法work 會出現 Failed precondition: Attempting to use uninitialized value batch_normalization_61/beta 想請問你的執行環境,或者有沒有人在rtx或tensorflow-gpu 1.14上執行過。 .h5 及 .pb均有重新convert過。

yehengchen commented 5 years ago

hi 我的環境是 rtx2070, cuda10, cudnn7.4, tensorflow-gpu 1.14 但無法work 會出現 Failed precondition: Attempting to use uninitialized value batch_normalization_61/beta 想請問你的執行環境,或者有沒有人在rtx或tensorflow-gpu 1.14上執行過。 .h5 及 .pb均有重新convert過。

你好, 只要把 .weights 文件转换成 .h5 文件就行了.pd 文件不需要 convert 直接导入就行,下面是我的测试环境配置: GeForce GTX 1080 Ti / cuda 9.0 / cuDNN v7.1.4 / tensorflow-gpu 1.10.0 / python 3.5 GeForce RTX 2080 Ti / cuda 10.0 / cuDNN v7.5.0 / tensorflow-gpu 1.13.1 / python 3.5 执行环境可以参考:https://github.com/yehengchen/Ubuntu-16.04-Deep-Learning-Environment-Setup/blob/master/README_CN.md