【问题标题】:Keras in PyCharm not using GPUPyCharm 中的 Keras 不使用 GPU
【发布时间】:2019-04-27 04:12:21
【问题描述】:

这些线程没有解决我的问题: Keras does not use GPU on Pycharm having python 3.5 and Tensorflow 1.4

Keras with TensorFlow backend not using GPU

我已经在运行 Windows 10 并具有 GTX 750 Ti 显卡的 PC 上安装了 Tensorflow 和 Tensorflow-gpu (v.1.12.0),因此它确实支持 CUDA。我还安装了 CUDA Toolkit v10 和 cuDNN 库,当我在命令提示符下运行 nvcc -V 时,我得到:

nvcc:NVIDIA (R) Cuda 编译器...

我正在使用 PyCharm,在 CPU 上运行 Keras 没有任何问题。但它不使用我的 GPU。

当我输入时

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

它说

2018-11-25 10:47:57.448275: I tensorflow/core/platform/cpu_feature_gaurd.cc:141] 您的 CPU 支持未编译此 TensorFlow 二进制文件以使用的指令:AVX2

[ ]

我尝试过的:

1) 我尝试卸载 Tensorflow 和 Tensorflow-gpu 并重新安装 Tensorflow-gpu,如上述线程所述。没有工作,我的代码不再在 CPU 上运行,并给出了关于 Tensorflow 的错误。一旦我重新安装了 Tensorflow,就又好了。

2) 我尝试使用一个名为 Theano 的库,它在 Keras 官方文档 here 中有所提及。当我添加以下几行时

import theano
theano.config.device='gpu'
theano.config.floatX='float32'

它说

警告 (theano.configdefaults): g++ 不可用,如果使用 conda: 'conda install m2w64-toolchain' C:\Users\HOME-PC\PyCharmProjects\env\lib\site-packages\theano\configdefaults.py.=:560: UserWarning: DeprecationWarning: there is no c++ compiler.This is deprecated and with theano 0.11 a c++ compiler will是强制性的。 warning.warn("DeprecationWarning: 没有 c++ 编译器。"

3) 我尝试添加这些行,但也没有任何反应。

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0" 

我在这里缺少什么?我应该如何将 CUDA 引入 PyCharm?

【问题讨论】:

  • 所以卸载 tensorflow 和 keras 没有帮助?
  • 但是如果它无法加载本机运行时,那没关系,这就是我们想要的。本机运行时是 CPU 之一。
  • 所以如果你没有 TF,只有 TF-gpu,你会遇到 keras 失败吗?您是否尝试过直接使用 TF keras 层而不是 keras?
  • 他们在tensorflow.keras,所以你应该可以像以前一样使用from TensorFlow import keras和使用API​​(tensorflow.org/guide/keras)。如果你使用这个API,你可以拥有tensorflow
  • 您确定您“完全”安装了 tensorflow 1.4 所需的 Cuda 版本吗? Tensoflow 很挑剔,它“完全”需要一个兼容的版本。我建议你去最新的 tensorflow 并安装教程中说的确切的 Cuda 版本。 -- 你不能同时拥有 tensorflow 和 tensorflow-gpu。仅安装 tensorflow GPU(如果您使用的是 conda,请尝试安装 keras-gpu)

标签: tensorflow keras pycharm


【解决方案1】:

可能并非如此,但安装和导入同一个库有时会让人感到困惑。 更清楚地说,我的猜测是你的 pycharm 环境与默认的 python 环境不同,你在默认环境中安装包并从 pycharm 的环境中导入。 为了确保你在pycharm的环境中安装了一个包,

您可以尝试以下方法, 来自pycharm的python控制台

!python -m pip install --upgrade tensorflow, keras

这将正确安装软件包,并且您可以确定安装没有错误

【讨论】:

  • 我假设你的意思是 tensorflow-gpu。但我也尝试从 PyCharm 项目解释器安装软件包。没有变化。
  • 您按照官方安装指南操作了吗?也许再次检查这些步骤可以帮助link
【解决方案2】:

问题出在 CUDA 的版本上。我已经安装了 CUDA v10.0,但 Tensorflow 似乎只适用于 v9.0。安装它,它就像一个魅力。

【讨论】:

    猜你喜欢
    • 2018-05-18
    • 1970-01-01
    • 1970-01-01
    • 2020-06-13
    • 2018-03-14
    • 1970-01-01
    • 2019-08-21
    • 2018-05-07
    • 2019-01-04
    相关资源
    最近更新 更多