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就对了。