先日、TensorFlow 2.0が正式にリリースされました。 この記事では、GPUサポートを有効にしたTensorFlow 2.0のDockerコンテナでJupyterを起動し、チュートリアルを動かすまでの手順を紹介します。 主な手順: Dockerのインストール NVIDIAドライバのインストール NVIDIA Container Toolkitのインストール TensorFlowコンテナの起動 Jupyterにアクセス チュートリアルで動作確認 主な環境: OS: Ubuntu 18.04.3 LTS GPU: GeForce GTX 1050 Ti やり方 Dockerのインストール NVIDIA Container ToolkitがDocker 19.03を求めているので、バージョンが古い場合はアップデートしてください。 新規インストールする場合は次のコマンドを実行します。 $