GPhilo
2020-03-12 23:32
Tensorflow 2.1+
这是怎么回事？
在新的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.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的日志记录级别来抑制警告，但这可能会过大，因为它将使所有警告静音。
Tensorflow 1.X或2.0：
您的CUDA设置已损坏，请确保安装了正确的版本。
热门帖子
相关问题
热门github
1
2
4
The Python Risk Identification Tool for generative AI (PyRIT) is an open access automation framework to empower security professionals and machine learning engineers to proactively find risks in their generative AI systems.
(翻译：用于生成式 AI 的 Python 风险识别工具 (PyRIT) 是一个开放式访问自动化框架，使安全专业人员和机器学习工程师能够主动发现其生成式 AI 系统中的风险。)
5
7
8
9
Mamba is a new state space model architecture showing promising performance on information-dense data such as language modeling, where previous subquadratic models fall short of Transformers. It is based on the line of progress on structured state space models, with an efficient hardware-aware design and implementation in the spirit of FlashAttention.
(翻译：Mamba 是一种新的状态空间模型架构，在信息密集型数据（例如语言建模）上显示出良好的性能，而之前的二次模型在 Transformers 方面存在不足。它基于结构化状态空间模型的进展，并本着FlashAttention的精神进行高效的硬件感知设计和实现。)
10
11
13