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 件のコメント:
コメントを投稿