DockerでGPUを活用する方法

Step-by-Step Guide to Leveraging GPU Power within Docker Containers

TL;DR Docker コンテナ内で GPU(nvidia) を使用したいという要望に応えるため、GPU 対応の Docker 設定を行う必要がある。 通常、Docker コンテナはホストの GPU リソースにアクセスできない。これは、Docker がホストの GPU デバイスをコンテナにパススルーしないからであり、GPU ドライバと CUDA ライブラリがホストとコンテナの間で適切に共有されていないのが原因である。 これを解消するには、nvidia-container-toolkit などのツールで GPU リソースをコンテナにパススルーする必要がある。 [続きを読む]
カテゴリー: docker  Tags: GPU nvidia