事の発端
何でもかんでも新しいことをしてみたくなる私は研究室の自分のUbuntuのバージョンもすぐに16.04から18.04にアップデート!
が、しかし!!!
教授「あ、これ18.04になってるからgpuが上手く使えてませんね。cudaを入れ直す必要がありますよ...かなり面倒ですが」
私「...か、かなりですか!?」
教授「はい、かなり面倒ですが...やりますか?(ニッコリ)」
私「...はい(涙目)」
ってことで、ちょっと色々調べてやったので、メモを残しておきます。
ステップ1
NVIDIA, CUDA, cuDNNのアンインストール。
自分のPCにはcuDNNが入って無かった?っぽいのでアンインストールしたのは前者2つのみ!
sudo apt-get --purge remove nvidia-*
sudo apt-get --purge remove cuda-*
ステップ2
CUDAをここからダウンロード。
Ubuntu18.04だとcuda-10.0らしいよ?
※教授曰く、debファイルよりrunファイルを使用した方が楽だそうで、私はrunファイルをダウンロードしました。
ステップ3
リポジトリの登録?
これは必要なのかはわからないけど、自分は一応しておいた。
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
ステップ4
Ubuntuの再起動!
ステップ5
ステップ2でダウンロードしてきていたファイルを実行する。
sudo ./cuda_10.0.130_410.48_linux.run
何か色々出てくるけど、とりあえずYesかAcceptしておけばOK...だと思うよ?笑
ステップ6
ステップ5でNVIDIAとCUDAがインストールされた筈なので次にcuDNNのインストールに入る。
cuDNNをインストールするにはここから登録してダウンロードする必要がある。
俺がダウンロードしたファイルは以下の3つ。
libcudnn7_7.5.1.10-1+cuda10.0_amd64.deb
libcudnn7-dev_7.5.1.10-1+cuda10.0_amd64.deb
libcudnn7-doc_7.5.1.10-1+cuda10.0_amd64.deb
ステップ7
ステップ6でダウンロードした3つをインストール(sudo dpkg -i ○○○
)する。
※上記に記載しているファイルの順番にインストールすること!
間違ってdevやdocを先にインストールしないように!!!
ステップ8
cuDNNがきちんとインストールできているのか以下の手順でサンプルを試してみる。
1.サンプルをホームディレクトリにコピーする。
$ cp -r /usr/src/cudnn_samples_v7/ $HOME
2. mnistCUDNNフォルダに入る。
$ cd $HOME/cudnn_samples_v7/mnistCUDNN
3. サンプルのmnistCUDNNをコンパイルする。
$ make clean && make
4.実行する。
$ ./mnistCUDNN
もし、何も問題が無ければ以下のようなメッセージが出ます。
Test passed!
ステップ9
終了!
お疲れ様でした〜!