ubuntu16.04+cuda8.0+cudnn5.1 安装

环境

Ubuntu16.04 cuda8.0 cudnn5.1

利用以下命令安装支持GPU的tensorflow

pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.1.0-cp27-none-linux_x86_64.whl(中文社区的版本都特别低,此处需要注意)

如果遇到权限问题,请使用完整路径

sudo -H xxxx/anaconda2/bin/pip install xxxxxx.whl
https://pypi.python.org/pypi/tensorflow 下载最新即可

这里写图片描述

OK~大功告成!
这里写图片描述
显示调用GPU:0.

PS:官网中提示的配置 TensorFlow 的 Cuba 选项
从源码树的根路径执行:
xxxxxx
这个地方已经配置过了应该,就是两个库和默认路径。所以可以不管这里。

测试neural style
https://github.com/anishathalye/neural-style
这里写图片描述

感动得要哭,确实比CPU要快很多!~

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值