簡體   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