【发布时间】:2018-07-14 17:10:09
【问题描述】:
我正在使用 pytorch 框架训练网络。我的电脑里有 K40 GPU。上周,我在同一台计算机上添加了 1080。
在我的第一个实验中,我在两个 GPU 中观察到相同的结果。然后,我在两个 GPU 上尝试了我的第二个代码。在这种情况下,我在 K40 中“不断地”获得了不错的结果,而在“完全相同的代码”中,在 1080 中获得了“不断地”糟糕的结果。
首先,我认为获得如此多样化输出的唯一原因是代码中的随机种子。所以,我这样固定种子:
torch.manual_seed(3)
torch.cuda.manual_seed_all(3)
numpy.random.seed(3)
但是,这并没有解决问题。我相信问题不可能是随机性的,因为我在 K40 中“不断”获得好结果,而在 1080 中“不断”获得坏结果。此外,我在其他 2 台计算机和 4 台其他 1080 GPU 上尝试了完全相同的代码,并且总是取得很好的结果。所以,问题一定是我最近插的1080。
我怀疑问题可能与驱动程序或我安装 pytorch 的方式有关。但是,我只在“一些”实验中得到不好的结果仍然很奇怪。对于其他实验,我得到了相同的结果。
谁能帮我解决这个问题?
【问题讨论】:
-
重新安装pytorch后问题似乎解决了!