tensorflow-gpu验证准确率是报错如上：
解决办法：
1. 加入os.environ['CUDA_VISIBLE_DEVICES']='2'
强制使用CPU验证-----慢
2.'batch_size', 降低为32，即可使用GPU跑------快
OP_REQUIRES failed at conv_ops.cc:386 : Resource exhausted: OOM when allocating tensor with shape..
本文介绍了解决TensorFlow-GPU在验证准确率时遇到的问题的方法。通过设置环境变量CUDA_VISIBLE_DEVICES或者调整batch_size大小，可以有效解决无法使用GPU进行验证的情况。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
