【问题标题】:Tensorflow can't find GPUTensorFlow 找不到 GPU
【发布时间】:2019-05-08 05:24:28
【问题描述】:

我很抱歉,因为有上百万个这样的问题,但似乎没有一个能回答我的问题。 我正在尝试重新安装 Tensorflow 以便它使用我的 GPU。

我在跑步:

  • Windows 10
  • Python 3.6.5
  • Keras 2.2.4
  • TF 1.13.1
  • Nvidia Quadro M1000M(驱动412.16,计算能力5.0)
  • CUDA 10.0.130
  • 已安装 CUPTI 并将 CuDNN 文件复制到 CUDA 文件夹中。

CUDA v10.0 文件夹、\extras\CUPTI\libx64 和 \include 文件夹都在我的 PATH 中,CUDAPATH 和 CUDA_PATH 都已定义)

然而,pip install tensorflow 它只能找到我的 CPU(使用:

from tensorflow.python.client import device_lib
device_lib.list_local_devices() 

如果我这样做,我似乎无法导入 tensorflow

pip uninstall tensorflow
pip install tensorflow-gpu
python
import tensorflow
ModuleNotFoundError: No module named 'tensorflow'

【问题讨论】:

标签: python tensorflow gpu


【解决方案1】:

尝试使用 --ignore-installed 选项再次安装 Tensorflow,例如:

pip install tensorflow-gpu==2.0.0-alpha0 --ignore-installed

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2020-03-12
    • 2021-03-21
    • 1970-01-01
    • 1970-01-01
    • 2022-10-15
    • 2018-08-30
    相关资源
    最近更新 更多