WSL2にnvidia-dockerを入れてベンチマークしてみた

数か月前に4Kテレビを購入したのをきっかけに、RTX3070を搭載したゲーミングマシンを購入しました。Windows11にアップデートするとWSL2のUbuntuでGPUが使えるようになったので、nvidiaのDockerランタイムを使ってどの程度のパフォーマンスが出るのか調べてみました。ベンチマークに使用したのは、以前の記事で使用したソースと同じで、比較のためほかにも3種類の環境で測ってます 。結果は以下の通り。

上から3つ目が今回WSL2のubuntuの上のdockerで動かした結果になります。２つ目と３つ目は同じPCで測定しているのですが、直接WindowsにインストールしたPythonで実行したときに比べて３倍以上遅くなってしまいました。環境構築方法がまずかったのか、あるいはWindows11が出たばかりでチューニングがされていないのか、それともやっぱりＷＳＬ２は遅いのか。原因はまだわかっていませんが、ずっと期待して待っていただけに残念です。

予想に反して健闘したのは、以前に購入したゲーミングPC（GTX970)です。Windows10からUbuntu20.04に変更したのですが、そのためか１秒だけ速くなりました。一方、Google Colaboratoryは前回測定時は4秒だったのに今回は7秒まで落ち込んでしまいました。割り当てられるGPUはその時によって違うらしいので、そのためかもしれませんが、こんなに違うものなのですね。ちなみに、今回ベンチマークに使用したノートブックはこちらです。