GPU 训练期间
作者: bencedm创建于 2025年5月6日更新于 2026年5月6日
亲爱的开发者们，我在使用 GPU 进行 Parametric UMAP 训练时遇到了一个问题：损失变成了 NaN，即使使用未经修改的官方示例 notebook 也会出现这个问题。我的设置如下： - GPU：NVIDIA GeForce RTX 4090 - CUDA/cuDNN：已检测到并正确加载 - TensorFlow 版本：我尝试了从 2.16 到最近的版本（目前是 2.19 吧） - 我使用了 umap-learn 的最新版本
内容来源: lmcinnes/umap