テンソルフローのCPUとGPUバージョンを同時にインストールすることは可能ですか

Question

ドッカーコンテナからGPUにアクセスするにはnvidia-dockerを使用しています。しかし、私たちのマシンのすべてがGPUを持っているわけではないので、GPUが利用できないときに自動的にCPUバージョンにフォールバックしたいと思います。

CPU用とGPU用に別々のドッカーイメージを作成する必要がありますか、またはtensorflowとtensorflow-gpuをインストールして、GPUが利用できるかどうかによって正しいバリアントを選ぶことができますか？

Answer 1

公式Tensorflow docker imagesを参照してください。GPUバージョンは、nvidia-dockerを使用してGPUにアクセスします。

私が過去に行ったことには、ほぼ同じ2つのDockefilesがあります。 、他の

すべてが同じになります（必要に応じあなたはまた、Python2の画像を選択することができます）

FROM tensorflow/tensorflow:latest-py3 

OR

FROM tensorflow/tensorflow:latest-gpu-py3 

：Dockerfilesの唯一の違いは、ディレクティブFROMだろうイメージをビルドするときに適切なFROMタグが設定されるように、これを自動化することもできます。私はホストマシンがGPUを持っているかどうかによってmakefileを使って適切なイメージを作りました。