deeplab源码:models/research/deeplab at master · tensorflow/models · GitHub
在本地仓库克隆下来即可。
注意!该源码是tensorflow1写的,装tensorflow2的话会需要改很多函数,建议用1!
可以在官网查看对应版本:在 Windows 环境中从源代码构建 | TensorFlow
我选的是python3.7+tensorflow1.15+cuda10+cudnn7.4的组合。
1. 在anaconda中创建环境,进入环境,安装以下包。
(jupyter在python3.7下用conda安装会报错,但是在3.6下不会)
>> conda create -n tf1_deeplab python=3.7
>> activate tf1_deeplab
>> conda install numpy
>> conda install pillow
>> pip install jupyter
>> conda install matplotlib
>> pip install tensorflow-gpu==1.15.0
2.安装对应版本的cuda和cudnn
cuda:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
(一路默认安装就行了)
cudnn: https://developer.nvidia.com/rdp/cudnn-archive
(下载下来是一个压缩包,把对应的文件放到cuda的相应的bin等目录下。)
3.测试tf是否安装成功,若返回true则成功
>> import tensorflow as tf
>> tf.test.is_gpu_available()
4.在pycharm里选择配置好的环境'tf1_deeplab'。
运行model_test.py测试。
报错:ModuleNotFoundError: No module named ‘tf_slim‘
解决:安装tf_slim
5.再次运行model_test.py,成功!