Tensorflow找不到GPU

时间:2019-05-08 05:24:29

标签: python tensorflow gpu

我很抱歉,因为这里有100万个问题,但似乎没有一个回答我。 我正在尝试重新安装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() 

如果我

,我似乎无法导入张量流
pip uninstall tensorflow
pip install tensorflow-gpu
python
import tensorflow
ModuleNotFoundError: No module named 'tensorflow'

1 个答案:

答案 0 :(得分:0)

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

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