tensorflow-gpu验证准确率是报错如上:
解决办法:
1. 加入os.environ['CUDA_VISIBLE_DEVICES']='2'
强制使用CPU验证-----慢
2.'batch_size', 降低为32,即可使用GPU跑------快
2024-03-22 18:50:28
tensorflow-gpu验证准确率是报错如上:
解决办法:
1. 加入os.environ['CUDA_VISIBLE_DEVICES']='2'
强制使用CPU验证-----慢
2.'batch_size', 降低为32,即可使用GPU跑------快
下一篇:模型剪枝