问题描述

跑机器学习代码时遇到这个问题

ImportError: Could not find ‘cudart64_90.dll‘. TensorFlow

原因

安装了错误的CUDA版本,例如本人安装的是2.3的tensorFlow,但是又安装了9.2的CUDA,应该为9.0版本的CUDA才对。

解决办法

安装9.0CUDA,或者安装与9.2CUDA版本对应的TensorFlow。

CUDA下载链接:https://developer.nvidia.com/cuda-toolkit
CUDNN下载链接:https://developer.nvidia.com/cudnn

相关文章: