Dockerを使って爆速でGPUを設定する方法 - Qiita

はじめに 機械学習をやろうと思ったら、まずやってくるのが GPUの設定 ですよね。ドライバインストールしてCUDAとcudnnインストールして、、、。これが本当にめんどくさい！そんな方には Dockerを用いた設定をオススメします。マジで爆速です。しかも、PyTorchやTFによって環境を別で設定できますし、使わなくなったらその環境をすぐに削除できます。ひと通りカバーした記事があまりなかったので残しておきます。困っている方の参考になればと思います。 *ミスがありましたら、ぜひご指摘お願いします 手順 1. NVIDIA driver のインストール 2. Docker のインストール 3. NVIDIA container toolkit のインストール 1. NVIDIA Driverのインストール ドライバがインストールされているか確認 インストールされてなければ、何も表示されない