本来想作一篇 Ubuntu 16.04 Tensorflow-gpu 1.4.1+GTX1080 的博客的,发现下面这个博主写的很赞,顶一个。
https://blog.youkuaiyun.com/u010478564/article/details/78766702#commentBox
不过在这里我还要补充两点:
1. 在设置默认gcc版本过后最好重启一下,要不然最后运行TensorFlow时仍然出现自己的gcc还是5.4.0版本的。
2. pip install tensorflow-gpu 时后面应该跟上版本号,如:pip install tensorflow-gpu==1.4.1,否则默认安装的是1.0.1版本的