Tensorflow 2.0 目標識別（三）——TF2.0

安裝TF命令，gpu版本：

pip install tensorflow_gpu==2.1.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

安裝好後可能會報錯的幾個地方：

找不到dll，可能的地方CUDA的環境變量沒弄對，cudnn版本和CUDA不匹配。

cudart64_*.dll,類似100改101之類的，能自然匹配最好，不行就搜素後改下名字。運行GPU版本的時候會慢慢出現陸續的dll，全是I就對了。

另一個可能，VC_redist.x64 .exe安裝一下，缺c++引導之類的。

https://support.microsoft.com/en-us/help/2977003/the-latest-supported-visual-c-downloads

 

測試GPU是否安裝成功，新建個*.py, 輸入

import tensorflow as tf
import object_detection
version = tf.__version__
gpu_ok = tf.test.is_gpu_available()
print("tf version:",version,"\nuse GPU",gpu_ok)

出來版本號和GPU就對了。