在Ubantu上安装tensorflow
来源:互联网 发布:安卓7.0 ios10 知乎 编辑:程序博客网 时间:2024/05/21 07:46
安装之前,需要安装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 100sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150
这样默认python就是python3了。可以试试:
python --versionpip -V
2.安装tensorflow
一键安装即可:
pip3 install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple/
后面参数是换源,这样会比较快。本来换了阿里的,结果发现找不到tensorflow-gpu???(点进去看还是找得到的,搜不到可能是因为python的版本)
接下来就是等待。
成功与否在此一举。
如果失败了别慌,看看是什么原因,很可能是因为超时了。
没办法人家清华服务器也不盈利日常炸锅。
不过我运气好。没有超时。(好像输入sudo就容易超时。。。)
2分钟不到,就下载完了。接下来就可以实验:
3.测试tensorflow安装成功与否
输入:python进入python环境,然后就是标准的tensorflow测试啦:
import tensorflow as tfhello = tf.constant('Hello, TensorFlow!')sess = tf.Session()print(sess.run(hello))第三条命令会报出一堆信息。那堆信息里大概能看到你的显卡型号,还有显存神马的!
至此,tensorflow安装结束。
阅读全文