ブックマーク / qiita.com/TokyoMickey (1)

  • ChainerでGPUを使うと毎回結果が変わる理由と対策 - Qiita

    記事はChainerやTensorFlowでGPUを使うと毎回結果が変わる理由と対策 (まとめ)の詳細編であり、Chainerについて具体的な対策について検証します。 関連Issues(Chainer) 検証環境 Chainer v1.20.0 Python 3.5.2 CUDA 8.0 cuDNN 5.1 NVIDIA Tesla/GeForce いろいろ まずは乱数種の指定 GPUの演算誤差だけでなく乱数を用いていることも演算結果が安定しない原因です。 そこで乱数種を指定して発生する乱数を固定します。 乱数種の指定対象 Chainerの場合、下記の3種類の乱数に配慮する必要があります。 Python NumPy CuPy CuPyはPFN社謹製のGPU対応版NumPyと考えればよいでしょう。 参考実装 def set_random_seed(seed): # set Python r

    ChainerでGPUを使うと毎回結果が変わる理由と対策 - Qiita
  • 1