Ubuntu16.04+Tensorflow+CUDA8.0

注:此篇为源码编译方式安装Tensorflow,不过也可以通过其他方法,比如Anaconda,我没试过:)

一、参考资料:

http://blog.csdn.net/zhaoyu106/article/details/52793183/

https://sanwen8.cn/p/23d9QiU.html】这个测试Tensorflow的方式和上面那个不一样,我也只是测试”通过源代码编译TensorFlow使其支持GPU的方式”

http://m.blog.csdn.net/article/details?hmsr=toutiao.io&id=52658965&utm_medium=toutiao.io&utm_source=toutiao.io

二、事先装好Ubuntu16.04+CUDA8.0

三、cuDNN v5:利用GPU加速卷积神经网络,提高整个网络训练速度。

  • 拷贝库文件:

    sudo cp cuda/include/cudnn.h /usr/local/cuda/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

  • 配置环境变量:

    • 因没有~/.bash_profile文件,改为修改~/.bashrc,添加以下路径:

    export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64”
    export CUDA_HOME=/usr/local/cuda

  • 路径生效:

    source ~/.bashrc

  • 四、安装其他库

    • pip:

      sudo apt-get install python-pip python-dev

    • Bazel:(谷歌开源构建工具,类似于Make,用于构建Tensorflow)

      方法一:
      这里写图片描述

    貌似不能用,采用第二种方法:

    方法二:
    这里写图片描述

    五、安装Tensorflow(由于开头找到的资料是关于Tensorflow0.11的,所以下载该版本的Tensorflow,否则安装时会有其他参数)

    按照参考博客来就行,期间遇到一个问题:

    这里写图片描述

    注意第4步

    sudo pip install /home/*(你自己的用户名)/Desktop/tensorflow-0.10.0-cp2-none-any.whl,
    我替换成
    sudo pip install /tmp/tensorflow_pkg/tensorflow-0.11.0-py2-none-any.whl
    出现错误【TypeError: unsupported operand type(s) for -=: ‘Retry’ and ‘int’】。

    解决办法(估计需要python2,而不是python3):

    apt-get remove python-pip python3-pip
    sudo apt-get install python-pip python-dev

    六、按照博客地址http://blog.csdn.net/zhaoyu106/article/details/52793183/测试结果:

    这里写图片描述

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值