Tensorflow-gpu 1.9 不适用于 Pycharm答案

【问题标题】：Tensorflow-gpu 1.9 is not working on PycharmTensorflow-gpu 1.9 不适用于 Pycharm
【发布时间】：2019-01-04 17:39:15
【问题描述】：

pycharm中无法导入tensorflow，报如下错误：

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

我检查了这个 cudart 文件并将其添加到 %PATH%

它在 CLI 和 python shell 上完美运行。

我有

CUDA v9.0
cuDNN v7.0.5
tensorflow-gpu v1.9

回显 %PATH%

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64;

cudar64_90.dll目录：

07/27/2018  02:53 PM    <DIR>          .
07/27/2018  02:53 PM    <DIR>          ..
09/02/2017  03:45 PM           163,840 bin2c.exe
07/27/2018  02:28 PM    <DIR>          crt
05/16/2018  11:18 PM        55,161,856 cublas64_90.dll
09/02/2017  03:45 PM           347,136 cuda-memcheck.exe
09/02/2017  03:45 PM         3,930,112 cudafe++.exe
09/02/2017  03:45 PM         4,226,048 cudafe.exe
09/02/2017  03:46 PM           299,520 cudart32_90.dll
09/02/2017  03:46 PM           373,760 cudart64_90.dll
11/16/2017  07:51 PM       286,877,184 cudnn64_7.dll
09/02/2017  03:46 PM       131,197,952 cufft64_90.dll
09/02/2017  03:46 PM           199,680 cufftw64_90.dll
09/02/2017  03:46 PM         3,575,808 cuinj32_90.dll
09/02/2017  03:46 PM         4,495,360 cuinj64_90.dll
09/02/2017  03:45 PM         1,411,072 cuobjdump.exe
09/02/2017  03:46 PM        48,057,344 curand64_90.dll
09/02/2017  03:46 PM        75,222,016 cusolver64_90.dll
09/02/2017  03:46 PM        54,782,464 cusparse64_90.dll
09/02/2017  03:45 PM           246,784 fatbinary.exe
09/02/2017  03:46 PM         1,274,880 gpu-library-advisor.exe
09/02/2017  03:46 PM           205,824 nppc64_90.dll
09/02/2017  03:46 PM         9,744,384 nppial64_90.dll
09/02/2017  03:46 PM         3,953,664 nppicc64_90.dll
09/02/2017  03:46 PM         1,035,264 nppicom64_90.dll
09/02/2017  03:46 PM         7,291,392 nppidei64_90.dll
09/02/2017  03:46 PM        55,641,088 nppif64_90.dll
09/02/2017  03:46 PM        26,491,904 nppig64_90.dll
09/02/2017  03:46 PM         4,767,232 nppim64_90.dll
09/02/2017  03:46 PM        14,943,232 nppist64_90.dll
09/02/2017  03:46 PM           179,200 nppisu64_90.dll
09/02/2017  03:46 PM         2,629,120 nppitc64_90.dll
09/02/2017  03:46 PM         9,024,512 npps64_90.dll
05/16/2018  11:18 PM           241,664 nvblas64_90.dll
09/02/2017  03:45 PM           325,632 nvcc.exe
09/02/2017  03:45 PM               328 nvcc.profile
09/02/2017  03:45 PM        16,261,120 nvdisasm.exe
09/02/2017  03:46 PM        15,747,584 nvgraph64_90.dll
09/02/2017  03:45 PM         7,202,304 nvlink.exe
09/02/2017  03:45 PM         4,005,376 nvprof.exe
09/02/2017  03:45 PM           181,248 nvprune.exe
09/02/2017  03:46 PM         3,182,592 nvrtc-builtins64_90.dll
09/02/2017  03:46 PM        17,302,016 nvrtc64_90.dll
09/02/2017  03:46 PM                53 nvvp.bat
05/16/2018  11:16 PM         7,082,496 ptxas.exe

【问题讨论】：

你能在你的shell中做echo %PATH%并将输出粘贴到这里吗？还要将dir 的输出粘贴到它所在的目录上吗？只是想确认它确实在您声称的路径上。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64;
你能把它放在你的问题中吗？你能把dll所在目录的dir也粘贴进去吗？
你是如何安装 CUDA 的？如果你有 cudart64_90.dll，请检查 cuda 的 bin 目录。如果没有，那么我建议从网络安装程序安装 CUDA。
按照这个步骤，1）完全卸载使用exe（网络）安装的CUDA Toolkit 9.0 2）使用exe（本地）安装CUDA Toolkit 9.0 再次安装Anacondas。按照说明tensorflow.org/install/install_windows 激活 Tensorflow 环境并安装 pandas、numpy、matplotlib、Keras、scipy、sklearn 等。然后从 anaconda 提示符打开 jupyter notebook。

标签： python tensorflow pycharm cudnn

【解决方案1】：

查看tensorflow-1.9release branch，看起来他们使用的是CUDA 9.0和CUDNN 7.1.4

所以我认为你应该下载 CUDNN 7.1.4。希望问题能得到解决。

您可以通过运行tensorflow 的一位作者编写的方便的script 来检查一切是否正常。它会准确地告诉你什么是不正常的。

编辑：对于 CUDA 10.0，它是 CUDNN 版本 7.6.2.24-1（请参阅 here）。

【讨论】：