我刚刚通过安装了Tensorflow的最新版本,pip install tensorflow并且每当我运行一个程序时,都会收到日志消息:
pip install tensorflow
W tensorflow / stream_executor / platform / default / dso_loader.cc:55]无法加载动态库’cudart64_101.dll’;dlerror:找不到cudart64_101.dll
这不好吗?如何解决错误?
在新的Tensorflow 2.1版本中,默认的tensorflowpip包同时包含CPU和GPU版本的TF。在以前的TF版本中,找不到CUDA库会发出错误并引发异常,而现在该库会动态搜索正确的CUDA版本,如果找不到它,则会发出警告(开头的 W 代表对于警告,错误以 E (或 F 表示致命错误)并退回到仅CPU模式。实际上,警告之后,错误消息也作为信息消息写入日志中(请注意,如果警告级别更高,最低日志级别(默认值,您可能看不到信息消息)。完整日志为(强调我的):
tensorflow
2020-01-20 12:27:44.554767:W tensorflow / stream_executor / platform / default / dso_loader.cc:55]无法加载动态库’cudart64_101.dll’;dlerror:找不到cudart64_101.dll 2020-01-20 12:27:44.554964:I tensorflow / stream_executor / cuda / cudart_stub.cc:29] 如果您的计算机上未设置GPU,请忽略cudart dlerror。
2020-01-20 12:27:44.554767:W tensorflow / stream_executor / platform / default / dso_loader.cc:55]无法加载动态库’cudart64_101.dll’;dlerror:找不到cudart64_101.dll
2020-01-20 12:27:44.554964:I tensorflow / stream_executor / cuda / cudart_stub.cc:29] 如果您的计算机上未设置GPU,请忽略cudart dlerror。
如果您的计算机上没有启用CUDA的GPU,或者您不关心没有GPU加速,则无需担心。另一方面,如果您安装了tensorflow并希望GPU加速,请检查您的CUDA安装(TF 2.1需要CUDA 10.1 , 而不是 10.2或10.0)。
如果只想消除警告,则可以调整TF的日志记录级别来抑制警告,但这可能会过大,因为它将使 所有 警告静音。
您的CUDA设置已损坏,请确保已安装正确的版本。