为深度学习GPU导入tensorflow库时出错？

Question

我的笔记本电脑上装有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

上获得 来自https://developer.nvidia.com/rdp/cudnn-download
的

2. cuDNN 7.1.4 for CUDA工具包9.0

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”中有该文件，并且我的系统环境路径变量也已配置到该目录，还有什么问题？ / p>

Answer 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文件，并在那里进行一些调试。