问题描述:

安装 tensorfolw-gpu2.1.0 之后调用 显示 GPU False,不能调用GPU

在命令行中 import tensorflow 时显示  “ Could not load dynamic library 'cudart64_101.dll”

注:V:显示全部信息  D:显示调试信息 I:显示一般信息 W:显示警告信息 E:显示错误信息

在以前的TensorFlow版本中,找不到CUDA库会发出错误并引发异常,而现在该库会从动力学上搜索正确的CUDA版本,如果找不到,则会发出警告(开头的W代表对于警告,错误以E(或F表示致命错误)并退回到仅CPU模式。实际上,警告之后,错误消息也作为信息消息写在日志中(请注意,如果警告级别更高,最低日志级别(默认值,您可能看不到信息消息)

原因:

缺少 cudart64_101.dll 文件

由于电脑上安装的 CUDA10.2 版本,缺少了 cudart64_101.dll 文件,导致 tf 调用改文件时无法读取

           

           (官网上 tensorflow-gpu2.1.0 建议安装 CUDA10.1 使用,可能也是这个原因)                                                                   

           https://www.tensorflow.org/install/source_windows#tested_build_configurations 

  解决办法:

搜索下载一个 cudart64_101.dll  文件放在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin(路                    径若为自定义需更改为实际路径)文件夹里面

或者将 cudart64_100.dll 或者 cudart64_102.dll (未必是这两个名字,根据你安装的CUDA版本而定)改名                             为 cudart64_101.dll

 

GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2 PiperOrigin-RevId: 663726708 2 个月前
91dac11a This test overrides disabled_backends, dropping the default value in the process. PiperOrigin-RevId: 663711155 2 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐