为深度学习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=exlocal

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

3. pip安装tensorflow gpu

导入tensorflow时使用：

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

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