DockerでGPUを使ってTensorFlowやPyTorchを動かす - Qiita

はじめに 本記事では，DockerでGPUを使ってTensorFlowやPyTorchを動かすために， 環境構築手順と動かし方を説明します． 環境構築 基本的には公式のインストール手順通りで，ところどころ追加で必要なものを入れていきます． 最新バージョンとの違いなどがあるかもしれませんので，参考ページと差異がある場合は参考ページに従ってください． 動作確認はUbuntu18.04で行いました． NVIDIAドライバの更新 参考：ubuntuにCUDA、nvidiaドライバをインストールするメモ 現在入っているドライバを削除します． $ sudo add-apt-repository ppa:graphics-drivers/ppa $ sudo apt update $ ubuntu-drivers devices == /sys/devices/pci0000:00/0000:00:00