目前,由于3060显卡驱动版本默认>11.0,因此,其不能使用tensorflow1版本的任何接口,所以学习在tf2版本下的目标检测驱动是很有必要的,此配置过程同样适用于任何30系显卡配置tf2。
一般配置Anaconda比较简单,这里便跳过,选用的anaconda版本为Anaconda3-2020.11-Windows-x86_64,可以在清华镜像官网上下载。
1,配置安装conda
本次选用的tensorflow版本为2.4,cuda为11.0,cudnn为8.0,对应python为3.7
首先设置虚拟环境,输入如下指令:
conda create -n tensorflow24gpu python=3.7
进入环境:
conda activate tensorflow24gpu
离开环境:
conda deactivate
2,安装cuda与cudnn
输入如下指令安装cuda
conda install cudatoolkit=11.0
输入如下指令安装cudnn
conda install cudnn==8.0.5.39 -c conda-forge
如下输入指令安装tensorflow2.4.0
pip install tensorflow==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
之后执行测试程序,输入python test.py其中内容如下,输出GPUTrue则安装正常,test.py代码如下
import tensorflow as tf
version = tf.__version__
gpu_ok = tf.test.is_gpu_available()
print("tf version:",version,"nuse GPU",gpu_ok)
3,tensorflow目标检测API配置运行
去官网https://github.com/tensorflow/models/tree/archive下载models-archve并解压文件到本地硬盘下的某一个盘符下,这里我解压到D盘,注意,解压 路径不要有中文
配置protoc,下载网址[https://github.com/protocolbuffers/protobuf/releases],这里我下载的为3.4.0版本win32位,解压后(位置任意)。接着将bin文件夹下的protoc.exe复制到C:Windows即可
之后,在models-archiveresearch目录下打开命令行窗口,输入:
protoc object_detection/protos/*.proto --python_out=.
若无错误,程序将不返回任何信息
添加环境变量。在Anaconda安装目录AnacondaLibsite-packages添加一个路径文件,如tensorflow_model.pth,必须以.pth为后缀,写上你要加入的模块文件所在的目录名称
D:TensorflowModelsmodels-archiveresearch
D:TensorflowModelsmodels-archiveresearchslim
再将解压目录,D:TensorflowModelsmodels-archiveresearchobject_detectionpackagestf2下的setup.py文件复制到D:TensorflowModelsmodels-archiveresearch,并在research目录下执行
python -m pip install .
若执行成功如图上所示。
之后安装运行模型所需要的一些必要的库,执行以下指令:
pip install tf_slim
pip install scipy
pip install tf-models-official==2.4.0
需要注意,执行最后一个指令时,会默认给当前环境安装最高版本的tensorflow2.8环境并卸载安装的tensorflow2.4因此,在执行完最后的安装tf-models-official后需要重新安装tensorflow2.4
pip install tensorflow==2.4.0
其中红包部分是由于部分库包与tensorflow2.4环境不匹配,降级版本或删除即可,这里google-cloud-bigquery一般不会用,删除即可,执行以下指令
pip install grpcio==1.32.0
pip uninstall google-cloud-bigquery
pip install grpcio-status==1.32.0
完成上述步骤后,执行测试文件,在research目录下输入以下指令:
python object_detection/builders/model_builder_tf2_test.py
输出以下结果,则配置安装正确
4,tensorflow目标检测API图片运行测试
在tensorflow24gpu的环境下执行以下语句:
conda install nb_conda
去tensorflow-model的github网站下载1.13版本https://github.com/tensorflow/models/tree/r1.13.0,并解压 ,将models-r1.13.0researchobject_detection下的object_detection_tutorial.ipynb拷贝到models-archiveresearchobject_detection下
在research文件夹目录下运行命令:jupyter notebook,
进入网页界面点击new后修改运行环境为tensorflow24gpu,如下
接着在jupyter中打开object_detection文件夹,并单击object_detection_tutorial.ipynb运行试例文件并在中间执行几处更改
在imports中插入以下命令
os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'
下载一个他人训练好的权重文件https://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_2017_11_17.tar.gz
解压并放在一个位置,将文件地址添加在Variables处,在Variables将代码改为:
#What model to download.
#Path to frozen detection graph. This is the actual model that is used for the object detection.
PATH_TO_FROZEN_GRAPH = 'D:/TensorflowModels/ssd_mobilenet_v1_coco_2017_11_17/frozen_inference_graph.pb'
# List of the strings that is used to add correct label for each box.
PATH_TO_LABELS = os.path.join('data', 'mscoco_label_map.pbtxt')
中间PATH_TO_FROZEN_GRAPH的目录为下载解压的ssd_mobilenet_v1_coco_2017_11_17训练集解压的位置,在将Download Model一块删除即可
最后,由于tensorflow1与2的区别,需要将object_detection_tutorial.ipynb代码中的tf全部修改为tf.compat.v1
之后点击run all运行
成功的运行结果如下所示:
GPU使用情况如下所示:
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:tensorflow2.4与目标检测API在3060显卡上的配置安装 - Python技术站