【问题标题】:TensorFlow GPU doesn't work, How to install it?TensorFlow GPU 不工作,如何安装?
【发布时间】:2019-07-23 01:51:48
【问题描述】:

最近开始学习tensorflow,决定切换到GPU版本,因为速度快很多,但是我无法导入,总是报同样的错误。

我已经试过了:

  • 在 python 3.6.8、cuda 10 和 cuda 10 的最新 cuDNN 中通过 pip 安装它
  • 我尝试重新安装 python、CUDA 和 cuDNN
  • 尝试安装 Visual Studio 并安装了 CUDA 9 和 cuDnn
  • 我尝试安装最新的 Anaconda,在 python 3.6 中创建了一个“默认”环境和另一个(也在 3.5 中尝试过),在这两种情况下都使用 pip install tensorflow-gpu
  • 我最后一次尝试是按照 youtube 上的教程进行操作,我完全按照演示进行操作 (https://www.youtube.com/watch?v=KZFn0dvPZUQ)

我尝试的所有方法都返回相同的错误。

回溯:https://pastebin.com/KMEsZAmq

完整代码:https://pastebin.com/7tS0Rd5S(正在处理 CPU 版本)

    .                      

我的规格:

  • i5-8400
  • 8 GB 内存
  • GTX 1060 6GB
  • W10 家庭 x64

【问题讨论】:

  • 请不要使用外部链接进行追溯或代码。我们希望您在问题中添加尽可能多的代码,以便即使 URL 失效,问题仍然相关。

标签: python tensorflow importerror


【解决方案1】:

看看这里: https://www.tensorflow.org/install/gpu

Tensorflow 支持 CUDA 9.0,您需要降级 CUDA 或使用 tensorflow 的 docker 镜像之一: https://www.tensorflow.org/install/docker

通过 docker,它不会使用您的 CUDA 驱动程序

【讨论】:

    猜你喜欢
    • 2020-10-21
    • 2018-07-12
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2019-03-14
    • 1970-01-01
    相关资源
    最近更新 更多