美文网首页我爱编程
ubuntu server 下安装tensorflow

ubuntu server 下安装tensorflow

作者: Cindy小隐 | 来源:发表于2016-11-17 18:27 被阅读0次

    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
    

    相关文章

      网友评论

        本文标题:ubuntu server 下安装tensorflow

        本文链接:https://www.haomeiwen.com/subject/zoddpttx.html