TensorFlow+Faster-RCNN+Ubuntu 环境配置&代码运行过程

代码链接:
https://github.com/endernewton/tf-faster-rcnn

1. 前期准备

  • 安装虚拟环境
创建一个tf-faster-rcnn的python2.7的虚拟环境

conda create –n tf-faster-rcnn python=2.7

进入此虚拟环境

source activate tf-faster-rcnn

备注:

a. 如果需要退出此虚拟系统,执行以下命令

source deactivate

b. 在虚拟环境中安装的东西会放在此根目录下/home/USERNAME/.conda/envs/tf-faster-rcnn(创建的虚拟环境名字)

查看cuda的版本

cat /usr/local/cuda/version.txt

查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grepCUDNN_MAJOR -A 2

根据cuda 和 cudnn版本,到tensorflow 的官网上查对应的TF版本。我的cuda是8.0,cudnn是5.1的,所以要安装1.2.0

https://www.tensorflow.org/install/install_sources#common_installation_problems


2. 环境配置

    • 0
      点赞
    • 35
      收藏
      觉得还不错? 一键收藏
    • 3
      评论

    “相关推荐”对你有帮助么?

    • 非常没帮助
    • 没帮助
    • 一般
    • 有帮助
    • 非常有帮助
    提交
    评论 3
    添加红包

    请填写红包祝福语或标题

    红包个数最小为10个

    红包金额最低5元

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

    抵扣说明:

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

    余额充值