环境
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要快很多!~