我爱编程

ubuntu server 下安装tensorflow

2016-11-17  本文已影响0人  Cindy小隐

ubuntu上安装GPU版tensorflow并没有那么麻烦,只要装好几个依赖的东西就会非常顺利。

安装anaconda

先装anaconda的好处是可以使用anaconda中的一些包,最起码不用再手动去装pip了。建议不要用sudo去安装anaconda,不然后面用pip装东西比较麻烦。

下载cuda并安装

下载cuda需要注册,而且各个版本的cuda差异很大,一定要选择合适的cuda。


这里对cuda的安装描述的非常清楚,我就不再班门弄斧了。

安装cudnn

不安装的话载入tensorflow会报“找不到libcudnn.so”这个错误。

安装tensorflow

如下命令即可安装tensorflow:

pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl
  1. 安装过程中可能会出现“SSLError: The read operation timed out” 这个错误,这是因为响应的时间设置不够长造成的,可以在安装命令中加上“--default-timeout=100”;如果还出现这个错误,可以手动下载进行安装。
  2. 为避免上一次的安装对本次的安装造成影响,在安装命令中加上“ --ignore-installed”;(如果已经安装失败过一次,那本次的安装可能会报错Cannot remove entries from nonexistent)。
  3. 最终安装命令如下。
pip install --default-timeout=100 --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl
上一篇 下一篇

猜你喜欢

热点阅读