TensorFlowでGPUを使えるようにする(docker編）

dockerでもGPUが使える

Windowsではdocker上でGPUが使えないと思っていたので、しばらくDockerは避けていましたが普通に使えました。
すべてここに書いてあることの翻訳に過ぎないのですが、備忘録として書き残します。

githubレポジトリに書いてあるように、ホストOSにもドライバをインストールする必要があるので、それぞれ下の記事を参照してください

Windows
Linux

起動例

dockerのバージョンで使用するフラグが変わるので注意してください。

docker>=19.03

$ docker run -it --rm --gpus all tensorflow/tensorflow:2.9.1-gpu /bin/bash

--gpus all

docker<19.03

$ docker run -it --rm --runtime=nvidia all tensorflow/tensorflow:2.9.1-gpu /bin/bash

--runtime=nvidia