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

原因
安装了错误的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版本与TensorFlow版本不匹配导致的问题。具体表现为安装了2.3版本的TensorFlow,但CUDA版本为9.2,正确的CUDA版本应为9.0。解决方案包括安装正确版本的CUDA或对应版本的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
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
1585
1910

被折叠的 条评论
为什么被折叠?