安装 tensorflow-gpu 版本后,需要安装相应的 CUDA 和 cuDNN 

  注意版本问题:tensorflow-gpu 1.7以及之后的版本要安装 CUDA 8.0 以上的版本,tf 1.7 之下的安装 CUDA 8.0 之下的 ,安装 cuDNN 时要根据 CUDA 的版本来选择

tensorflow运行出现错误 : ImportError: Could not find 'cudart64_90.dll'.

 运行时错误如下:

tensorflow运行出现错误 : ImportError: Could not find 'cudart64_90.dll'.

  根据错误信息安装 CUDA 9.0 即可,注意是 9.0 不是9.2版本

CUDA 9.0 下载地址:https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork  下载完成后安装即可

cuDNN 7.0.5 下载地址:https://developer.nvidia.com/rdp/cudnn-archive

  1.下载 cuDNN 时需要注册英伟达的账号,建议使用 gmail 邮箱注册,本人使用了QQ 和网易两个邮箱都没能收到验证邮件,激活不了账号,

  2. cuDNN 下载完成后解压,将解压后的三个文件夹 bin,include,lib 复制到 CUDA 的安装路径下和重名的三个文件夹进行合并, CUDA 的默认安装路径为:

1 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0

tensorflow运行出现错误 : ImportError: Could not find 'cudart64_90.dll'.

复制到CUDA 路径 v9.0 文件夹下进行合并:

tensorflow运行出现错误 : ImportError: Could not find 'cudart64_90.dll'.

 

更多问题可参考:https://github.com/tensorflow/tensorflow/issues/17101