1 anaconda 安装
2 子环境创建 必须创建 会下载必要的支持的库
conda create -n tf2.0 python=3.7 //创建
conda activate tf2.0 //激活
3 用阿里源安装tf2.0
pip install tensorflow2.0.0 -i https://mirrors.aliyun.com/pypi/simple
pip install tensorflow-gpu2.0.0 -i https://mirrors.aliyun.com/pypi/simple
个人建议两个版本分开安装在单独的python环境里面
cuda 下载 10.2版本
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
cudnn 下载 v7.6.5 (November 18th, 2019), for CUDA 10.2
https://developer.nvidia.com/rdp/cudnn-download
4测试一下
打开spyder 创建一个python脚本如下
“”“
test.py
“”“
import tensorflow as tf
version = tf.version
gpu_ok=tf.test.is_gpu_available()
print(“tf version:”,version,"\nif use GPU:",gpu_ok)
5 下载Tensorflow object detection API
https://github.com/tensorflow/models
从github上下载项目(右上角“Clone or download”-“DownloadZIP”),下载到本地目录(避免中文),解压。
6 Protobuf 安装与配置
在 https://github.com/google/protobuf/releases 网站中选择windows 版本(最下面),解压后将bin文件夹中的【protoc.exe】放到C:\Windows
在models\research\目录下打开命令行窗口,输入:
From tensorflow/models/
protoc object_detection/protos/*.proto --python_out=.
在这一步有时候会出错,可以尝试把/*.proto 这部分改成文件夹下具体的文件名,一个一个试,每运行一个,文件夹下应该
出现对应的.py结尾的文件。不报错即可
7 PYTHONPATH 环境变量设置
在 ‘此电脑’-‘属性’- ‘高级系统设置’ -‘环境变量’-‘系统变量’ 中新建名为‘PYTHONPATH’的变量,将
models/research/ 及 models/research/slim 两个文件夹的完整目录添加,分号隔开,效果如下图:
接下来可以测试API,在 models/research/ 文件夹下运行命令行:
python object_detection/builders/model_builder_test.py
报错:
Traceback (most recent call last):
File “object_detection/builders/model_builder_test.py”, line 21, in
from object_detection.builders import model_builder
ImportError: No module named ‘object_detection’
解决:
在site-packages添加一个路径文件,如tensorflow_model.pth,必须以.pth为后缀,写上你要加入的模块文件所在的目录名称就是了,如下:
D:\Anaconda3\Lib\site-packages\tensorflow\models\research
D:\Anaconda3\Lib\site-packages\tensorflow\models\research\slim
PS.有人说把这两个地址加进环境变量,我试了,反正我这不好使.
问题
AttributeError: module ‘tensorflow’ has no attribute ‘contrib’
http://www.tensorfly.cn/tfdoc/api_docs/README1.html
通过 tf_upgrade_v2.exe(安装tf2.0时候自带)