Debian 9.3 ディープラーニング GPUドライバの導入 (Linux自作PC 14)

当サイトではアフィリエイト広告を利用しています

Debian 9.3 Stretchでディープラーニング用にGeforce 1070のGPUドライバをインストールします。



前回までに基本設定を行いました。いよいよGPUの設置と行きたいのですが、

本記事ではその前にGPUのドライバとライブラリのインストールを行っておきます。

導入するドライバ

今回私が購入したGPUはEVGA Geforce GTX1070 FTW2になります。

DebianのwikiにNvidiaのドライバに関する記述が載っています。

上記の内容を参考にすると、Debian9(Stretch)であり、Kelper/Maxwell以降の世代であればVersion375.66で問題ないそうです。supported deviceの一覧にGTX1070の記載もあるので大丈夫でしょう。

インストール

既にリポジトリにnon-freeとcontlibは追加済みであるものとします。

インストールの手順はDebian Wikiに従って行います。

$ sudo apt install linux-headers-$(uname -r|sed 's/[^-]*-[^-]*-//') nvidia-driver

インストールが完了したら再起動し、温度監視などにnvidia-smiをインストールしておきます。

$ sudo apt install nvidia-smi

追加インストール

ディープラーニング関連で後程で行うNvidia-Dockerの導入において、libcudaが足りていなくてエラー対処が必要でしたので、ここで先にインストールしておいた方がよいでしょう。

$ sudo apt install libcuda1 libcudart8.0

今後

GPUドライバの導入が終わったので、いよいよGPUの設置を行います。