安装之前,需要安装CUDA和cuDNN,见我另两篇配套文章:
CUDA安装
cuDNN安装
环境:(环境不一致也可以参考一下,但是不能无脑参考了)
Ubantu 16.04
Cuda 8GA2(没有安装update)
展开一个分支,因为python3会成为以后的主流,所以如果有小伙伴想使用python3的话,可以这么输入:
接下来我会切换到python3默认的模式(因为Ubantu默认输入python用的是python2的环境),输入:
这样默认python就是python3了。可以试试:
CUDA安装
cuDNN安装
环境:(环境不一致也可以参考一下,但是不能无脑参考了)
Ubantu 16.04
Cuda 8GA2(没有安装update)
1.安装其他库
其实就是pip啦。装好了就不用装了。
sudo apt-get install python-pip python-dev
少输入第二项后果自负。
展开一个分支,因为python3会成为以后的主流,所以如果有小伙伴想使用python3的话,可以这么输入:
sudo apt-get install python3-pip python3-dev
接下来我会切换到python3默认的模式(因为Ubantu默认输入python用的是python2的环境),输入:
sudo update-alternatives --install /usr/bin/python python /usr/bin/python2 100
sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150
这样默认python就是python3了。可以试试:
python --version
pip -V
2.安装tensorflow
一键安装即可:
pip3 install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple/
后面参数是换源,这样会比较快。本来换了阿里的,结果发现找不到tensorflow-gpu???(点进去看还是找得到的,搜不到可能是因为python的版本)
接下来就是等待。
成功与否在此一举。
如果失败了别慌,看看是什么原因,很可能是因为超时了。
没办法人家清华服务器也不盈利日常炸锅。
不过我运气好。没有超时。
超时了怎么办?
不断尝试!
不断尝试!
不断尝试!
我反正,另一台电脑试了十几次,就成功了。
至于有人说,加长服务器响应时间的,其实没用,因为成功的时候,是很快就下完了,失败的时候,卡着一辈子动不了。
2分钟不到,就下载完了。接下来就可以实验:
3.测试tensorflow安装成功与否
输入:python
进入python环境,然后就是标准的tensorflow测试啦:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
第三条命令会报出一堆信息。那堆信息里大概能看到你的显卡型号,还有显存神马的!
至此,tensorflow安装结束。