- 博客(2)
- 收藏
- 关注
RSS订阅原创 关于TensorFlow-gpu出现nan原因
当我使用CPU跑程序的时候，结果的LOSS正常输出，但是当我改成使用GPU时，我的loss显示为nan。但是我的输出图片是可以输出的，程序也是可以运行的，说明问题出现在loss函数这一部分。查阅资料后发现，TensorFlow-gpu版不支持双星号（**）运算，所以在损失计算时，返回的数组就会变成nan。于是我就去找我的loss的这一部分代码，发现确实有双星号的存在。于是我就把双星号去掉，改成最原始的样子（src-dst）*(src-dst)，结果loss就可以正常输出。有可能GP.
2022-03-27 20:57:56
1316
原创 使用GUP运行，结果没有变化
当我第一次使用GPU时，运行提示框一直不变化，但是我的GPU显示在运行，查了好多资料，没有太实质的作用。把电脑放在一边，过段时间在看，发现已经变化了，成功输出结果！！ 好像是因为第一次运行GPU过慢，大家可以多运行两次，等待一段时间，看看是否有变化。I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/job:localhost/replica:0/tas...
2022-03-27 20:45:01
497
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人