安装 tensorflow-gpu 版本后,需要安装相应的 CUDA 和 cuDNN
注意版本问题:tensorflow-gpu 1.7以及之后的版本要安装 CUDA 8.0 以上的版本,tf 1.7 之下的安装 CUDA 8.0 之下的 ,安装 cuDNN 时要根据 CUDA 的版本来选择
运行时错误如下:
根据错误信息安装 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
复制到CUDA 路径 v9.0 文件夹下进行合并:
更多问题可参考:https://github.com/tensorflow/tensorflow/issues/17101
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:tensorflow运行出现错误 : ImportError: Could not find ‘cudart64_90.dll’. - Python技术站