【发布时间】: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'
【问题讨论】:
-
在设置 PATH 变量后,您是否尝试过重新启动计算机?在您完全重新启动计算机之前,它们不会完全安装。
-
毕竟在这里找到了答案:看起来一些剩余的模块没有正确卸载。 stackoverflow.com/questions/42326748/…
-
有时你可以安装多个版本的 TF,你必须继续使用 pup uninstall tensorflow 直到它们都消失,然后使用 pip install tensorflow-gpu
标签: python tensorflow gpu