【问题标题】:Error while importing tensorflow library for deep learning GPU?为深度学习 GPU 导入 tensorflow 库时出错?
【发布时间】:2018-08-05 11:14:14
【问题描述】:

我的笔记本电脑上有一块 NVIDIA GeForce 940mx GDDR5 2GB 的 GPU。我想使用支持 GPU 的 TensorFlow。

我尝试了从链接安装 tensorflow 的步骤 https://www.tensorflow.org/install/install_windows

我已经安装了:

  1. CUDA 9.0 工具包以及所有三个补丁更新可在 https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

  2. 来自https://developer.nvidia.com/rdp/cudnn-download的 CUDA 工具包 9.0 的 cuDNN 7.1.4

  3. pip install tensorflow-gpu

在导入张量流时使用:

import tensorflow as tf

我遇到了一个错误:

ImportError: Could not find 'cudart64_90.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 9.0 from this URL: https://developer.nvidia.com/cuda-toolkit

我在 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin' 中有那个文件,并且我的系统环境路径变量也已配置到此目录,还有什么问题?

【问题讨论】:

标签: python tensorflow machine-learning cudnn


【解决方案1】:

我将在这里推测一下,所以也许我错了,但我会假设您已经运行了命令提示符,安装了 CUDA,更新了 env.var PATH 添加了 CUDA 路径你忘了重新启动命令提示符。如果是这样,您运行 Python 的进程不会更新 PATH env.var?要确定这一点,只需执行python -c 'import os; print(os.environ["PATH"])' 并确认即可。

如果 PATH 在命令提示符过程中正常,请再次检查 CUDA 目录,搜索 cudart64_90.dll 并确保该文件所在的路径已正确添加到 PATH。

如果上一步失败了,那么最好的机会是打开加载该 dll 的 Tensorflow 文件并在那里进行一些调试。

【讨论】:

    猜你喜欢
    • 2018-08-31
    • 2017-12-09
    • 1970-01-01
    • 2017-09-27
    • 1970-01-01
    • 2020-11-10
    • 1970-01-01
    • 2019-10-24
    • 1970-01-01
    相关资源
    最近更新 更多