【问题标题】:downgrade tensorflow to use cuda 8降级 tensorflow 以使用 cuda 8
【发布时间】:2018-07-06 12:25:11
【问题描述】:

当我运行 tensorflow 时,我得到:

ImportError: libcublas.so.9.0: cannot open shared object file: No such 文件或目录

所以我想将我的 tensorflow-gpu 降级为使用 cuda 8。我该如何使用 pip 来做到这一点,我应该下载哪个版本?

【问题讨论】:

    标签: tensorflow


    【解决方案1】:

    使用1.4.1版本的tensorflow作为

    pip install tensorflow-gpu==1.4.1
    

    【讨论】:

    • 我有 cuda9.1 然后我删除了它并安装了 cuda8 并且 tf 仍然想要libcublas.so.9.1。现在,我有 tf 1.4.1、Cuda 8 和 cudnn7。我必须降级cudnn吗? tensorflow.org/install/…
    • tensorflow 1.4.1的官方文档建议使用CuDNN 6.0 你可以尝试使用最新的CuDNN 7.1,如果不行你应该降级到CuDNN 6.0
    • 我正在使用cudnn 7.1.4。从 tf 文档tensorflow.org/versions/r1.5/install/… 我认为 cudnn 7 应该可以工作。另外,libcublas.so.9.1 问题是否与 cudnn 版本有关?
    • 是的,你需要有 CUDA 8.0 的 cudnn,这就是你正在使用的
    • 但是在 NVIDIA 网站上,Cuda 8 与 cudnn 7.1.4 兼容 developer.nvidia.com/rdp/cudnn-download 你的意思是 cudnn7.1 不能在 tensorflow1.4 上工作吗?
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2020-06-13
    • 1970-01-01
    • 2021-03-20
    相关资源
    最近更新 更多