More than 3 years have passed since last update.

chainer+CUDA(11.0)+cuDNN(8.0)で畳み込みする際にPerformanceWarningが出る

Last updated at 2021-02-20Posted at 2021-02-20

概要

chainerでGPUを使ってCNNをする際に，以下のようなWarningが出てきました．

PerformanceWarning: 
The best algo of conv fwd might not be selected due to lack of workspace size (8388608)

実行自体は問題ないのですが，Warningが出るのは気持ちが悪いので修正方法を探しました．

同じような問題に遭遇している方がいらっしゃいました．
chainerでTensorcoreを使って学習の高速化

どうやらcuDNNがGPUのメモリ(workspace)を確保する際にサイズが不足していることが原因らしいです．

解決するには以下のようにworkspaceのサイズを設定することで，必要なメモリサイズを確保してあげれば良いそうです．

ws_size = 256 * 1024 * 1024 # 256MB
chainer.cuda.set_max_workspace_size(ws_size)

この例の場合256MBを確保しています．
確保するサイズを調整することによっていづれPerformanceWarningが出なくなります．

私の場合は512MBを確保したらPerformanceWarningが出なくなりました．

他の複数の環境では出たことがなかったので環境依存なんですかね？