博主在尝试在配备y9000p、Win11系统和3060显卡的电脑上安装不同版本的Python、CUDA和cuDNN以及tensorflow-gpu时遇到loss为nan或数值异常的问题。即使使用CPU运行tensorflow或者在服务器上使用GPU运行都是正常的，这排除了学习率、模型和数据集的潜在问题。经过一天的排查，最终发现问题是由于Win11系统或显卡与CUDA的兼容性导致。解决方法是将环境配置为python=3.8、CUDA=11.3、cuDNN=8.2.1和tensorflow-gpu=2.7.0，问题得到解决。
首先说下我的电脑是有y9000p，win11系统，3060显卡
之前装了好几个版本都不行 。
python=3.6 CUDA=10.1 cuDNN=7.6 tensorflow-gpu=2.2.0或者2.3.0
python=3.8 CUDA=10.1 cuDNN=7.6 tensorflow-gpu=2.3.0
都出现了loss一直为nan，或者loss，accuracy数值明显不对的问题
尝试了一下用CPU tensorflow跑是正常的，并且也在服务器上用GPU跑了显示正常，说明不是学习率，网络模型，数据集等的问题。
从上图可以看到loss和accuracy的值明显是有问题的
查找了一天博客也没找到解决方法。最后试了下重装了一个以下配置的环境，问题解决成功运行。
python=3.8 CUDA=11.3 cuDNN=8.2.1 tensorflow-gpu=2.7.0
分析了一下问题可能是win11系统的问题，也可能是显卡与CUDA的兼容问题
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
2735
337
413
2307
659
1958
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
