TensorFlow运行报cudart64_110.dll not found,已装CUDA10.1仍未解决怎么办
问题修复方案
根本原因
报错核心为版本不匹配:TensorFlow 2.6.0 官方要求配套的 CUDA 版本为 11.2、cuDNN 版本为 8.1,你当前环境安装的 cudatoolkit 10.1 版本不符合要求,因此程序查找 CUDA 11 对应的cudart64_110.dll文件失败。
可选修复方案
方案1:保留现有CUDA 10.1,降级TensorFlow到适配版本
TensorFlow 2.4.x 版本适配 CUDA 10.1,操作步骤如下:
- 激活目标conda环境:
conda activate myEnv001 - 卸载现有TensorFlow:
pip uninstall tensorflow - 安装适配版本的TensorFlow:
pip install tensorflow==2.4.0
方案2:保留TensorFlow 2.6.0,安装匹配的CUDA/cuDNN版本
- 激活目标conda环境:
conda activate myEnv001 - 卸载现有低版本cudatoolkit:
conda remove cudatoolkit - 安装匹配版本的依赖:
conda install -c anaconda cudatoolkit=11.2 cudnn=8.1
补充说明
如果仅需使用CPU运行TensorFlow,不需要GPU加速,可以在代码开头导入TensorFlow前添加以下代码隐藏报错:
import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
如果执行上述方案后仍有报错,可以手动将conda虚拟环境的库路径添加到系统PATH中,路径一般为你的conda安装目录/envs/myEnv001/Library/bin。
内容的提问来源于stack exchange,提问作者H42
相关产品推荐
相关产品推荐

