繁体   English   中英

Tensorflow找不到GPU

[英]Tensorflow can't find GPU

我很抱歉,像这样有上百万个问题,但似乎没有一个回答我。 我正在尝试重新安装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'

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

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

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM