2017年3月23日木曜日

CentOS7にcuDNNをインストール

(参照http://qiita.com/skanai/items/0f0a9bdd7e04d55db981)
cuDNNは、Deep Neural NetworkをGPUで高速化するためのライブラリである。NVIDIAが提供する。インストール作業は手間じゃないけど、cuDNNをダウンロードするとき、NVIDIAのサイトでログインする必要がある(アカウントのない人は登録する)。

https://developer.nvidia.com/cudnn

ダウンロードしたディレクトリで解凍する。
$ tar xvfz cudnn-8.0-linux-x64-v5.1.tgz

解凍された一連のファイルを、CUDAをインストールしたディレクトリにコピーする。
$ cd cuda
$ sudo cp -a lib64/libcudnn* /usr/local/cuda/lib64/
$ sudo cp -a include/* /usr/local/cuda/include/
$ sudo ldconfig

0 件のコメント:

コメントを投稿