您的位置:首页 > 其它

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

2017-05-08 21:18 447 查看

环境

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(中文社区的版本都特别低,此处需要注意)[/code] 
如果遇到权限问题,请使用完整路径

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要快很多!~
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐