Ubuntu16.04 部署 TensorFlow2-GPU版本(虚拟环境)


部署tf2虚拟环境

一、创建虚拟环境(Python=3.6.5)

conda create -n tf2 python=3.6.5

激活虚拟环境,使用如下命令:

source activate tf2

二、安装CUDA、CuDNN环境

conda install cudatoolkit=10.1 cudnn=7.6.5

三、安装指定版本tensorflow

pip install tensorflow-gpu==2.3.4

附:最开始因为pip的版本太低了,没有找到对应的安装包,使用“pip install --upgrade pip”对pip进行升级后,重新执行上述命令,安装成功。下图为报错信息:

四、验证是否安装成功

python
import tensorflow as tf
print(tf.__version__)

显示结果参考下图:

验证GPU是否可用:

import tensorflow as tf
tf.config.list_physical_devices(GPU)

显示结果参考下图:


部署目标检测API

下载models文件到指定目录:

在指定目录下,运行下列命令:

​git clone https://github.com/tensorflow/models.git

protobuf 安装/编译:

protoc object_detection/protos/*.proto --python_out=.

此时会在指定目录,出现一个models目录

然后执行下述命令:

cd /models/research/object_detection/packages/tf2
​​​​​​​python -m pip install .

附:该命令会自动安装tf2所需要的相关依赖库,避免我们在使用时逐个安装.

经验分享 程序员 微信小程序 职场和发展