日々精進

新しく学んだことを書き留めていきます

GPUでTensorFlowを実行出来るDockerコンテナを作る

基本は以下の記事に書いてあるDockerfileで大丈夫だが、このままだとGPUが使えないという問題がある。

anton0825.hatenablog.com

GPUを使えるようにするにはベースイメージのところを FROM nvidia/cuda:9.0-base-centos7 に変更する。

このとき、cuda9.0のbase imageを使うのがポイントで9.1以降だとGPUが使えない。

nvidia/cudaのDockerHubではUbuntuとCentOSのイメージが用意されているのでどちらかのOSを使っている場合はこれらをbase imageにするといい。

hub.docker.com