在新的Tensorflow
2.1版本中,默认的
tensorflowpip包同时包含CPU和GPU版本的TF。在以前的TF版本中,找不到CUDA库会发出错误并引发异常,而现在该库会动态搜索正确的CUDA版本,如果找不到它,则会发出警告(开头的
W 代表对于警告,错误以 E (或 F
表示致命错误)并退回到仅CPU模式。实际上,警告之后,错误消息也作为信息消息写入日志中(请注意,如果警告级别更高,最低日志级别(默认值,您可能看不到信息消息)。完整日志为(强调我的):
我应该担心吗?我如何解决它?2020-01-20 12:27:44.554767:W tensorflow / stream_executor / platform /
default /
dso_loader.cc:55]无法加载动态库’cudart64_101.dll’;dlerror:找不到cudart64_101.dll2020-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的日志记录级别来抑制警告,但这可能会过大,因为它将使 所有 警告静音。
Tensorflow 1.X或2.0:
您的CUDA设置已损坏,请确保已安装正确的版本。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)