(win10)deeplab环境安装:tensorflow1.15+cuda10+cudnn7.4

deeplab源码:models/research/deeplab at master · tensorflow/models · GitHub

在本地仓库克隆下来即可。

注意!该源码是tensorflow1写的,装tensorflow2的话会需要改很多函数,建议用1!

可以在官网查看对应版本:在 Windows 环境中从源代码构建  |  TensorFlow

(win10)deeplab环境安装:tensorflow1.15+cuda10+cudnn7.4_第1张图片

我选的是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

(一路默认安装就行了)

(win10)deeplab环境安装:tensorflow1.15+cuda10+cudnn7.4_第2张图片
cudnn: https://developer.nvidia.com/rdp/cudnn-archive

 (下载下来是一个压缩包,把对应的文件放到cuda的相应的bin等目录下。)

(win10)deeplab环境安装:tensorflow1.15+cuda10+cudnn7.4_第3张图片

(win10)deeplab环境安装:tensorflow1.15+cuda10+cudnn7.4_第4张图片

 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

(win10)deeplab环境安装:tensorflow1.15+cuda10+cudnn7.4_第5张图片

 5.再次运行model_test.py,成功!

(win10)deeplab环境安装:tensorflow1.15+cuda10+cudnn7.4_第6张图片

 

你可能感兴趣的:(个人,tensorflow,深度学习,人工智能)