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

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,成功!

 

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值