问题描述
跑机器学习代码时遇到这个问题
原因
安装了错误的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
跑机器学习代码时遇到这个问题
安装了错误的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