WindowsのDockerでGPUが使えるようになっている…?


はじめに

個人的メモです。
最新の状況はどうかなと思うて検索したところたまたま見つかった記事でして、試せてはいません。
詳しくは下記リンクから。
https://docs.microsoft.com/ja-jp/virtualization/windowscontainers/deploy-containers/gpu-acceleration

出来る(らしい)こと

  • Docker Desktop で DirectX を通した GPU 利用ができる

TensorFlow などは無理そうかな…。

システム要件

  • Windows Server 2019 または Windows 10 (1809以降)
  • mcr.microsoft.com/windows:1809 以降のコンテナーイメージ
  • 19.03 以降の Docker エンジン
  • WDDM2.5 以上に対応したグラフィックドライバー

環境の起動方法

docker run --isolation process --device class/5B45201D-F2F2-4F3B-85BB-30FF1F953599 mcr.microsoft.com/windows:1809

デバイスを指定している以外、シンプルなコマンドですね。

所感

不勉強なので使えそうなものがC++/AMPぐらいしか思いつかないのですが、これまでの状況を考えるとなかなかの進展といえるのでは。