この記事の内容
tensorflowで機械学習するために、tensorflowのバージョンに対応するCUDAとcuDNNをローカルでインストールした時の作業を記録した。
本記事はtensorflow 2.3.0を使用するために、CUDA10.1とcuDNN7.6のセットアップを行った。
環境
Ubuntu 20.04
参考サイト
https://codelabo.com/posts/20200229081221
https://medium.com/@exesse/cuda-10-1-installation-on-ubuntu-18-04-lts-d04f89287130
本記事の構成
- 1:NDIVIAドライバーインストール編
- 2:CUDAセットアップ編
- 3:cuDNNセットアップ編
cuDNNセットアップ編
ダウンロードサイト:https://developer.nvidia.com/cudnn
事前にnvidiaのアカウント登録が必要。ダウンロードサイトのアーカイブからバージョンが選択できる。
1.cuDNN7.6の選択
- Download cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.1
- cuDNN Library for Linux
2.tarファイルからセットアップ
公式guideを参考にし順次実行していく。
2.1.解凍
$ tar -zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
2.2.事前にインストールしたcudaにcudnnをコピーする
cudnnをcuda/かcuda-<version>/のどちらかにコピーする。公式ドキュメントにはcuda/にしているが、cuda-<version>/とcuda/がリンクされているらしくcuda-<version>/にコピーしてもうまくいった。
# cuda-<version>/にコピーする場合
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda-<version>/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-<version>/lib64
$ sudo chmod a+r /usr/local/cuda-<version>/include/cudnn*.h /usr/local/cuda-<version>/lib64/libcudnn*
# cuda/にコピーする場合
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
入れたCUDAのバージョンを<version>にいれてください。
tensorflowのバージョンの変更に伴ってcudaとcudnnのバージョンを変更する場合
欲しいバージョンのcudaをインストールしcudaのパスを変更or追記
事前にCUDAをインストール。CUDAセットアップ編を参照してください。
追記した場合、自動的に最適なCUDAのバージョンのPATHを選んで実行してくれるらしい。
参考:https://qiita.com/takeajioka/items/8737fab5cffbe0118fea
実際いろんなやり方が存在する。
~/.bashrcに記述したパスを変更する。
例
<変更前 10.1>
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
<変更後 11.0>
export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
cudnnの変更
cuda-<version>/にcudnnをコピーした場合
pcにロードされているcudaのバージョンを切り替えると自動的に変更後のcudaにコピーされているcudnnのバージョンに切り替わると思う。(まだやってないが)
cuda/にcudnnをコピーした場合
/usr/local/cuda/include
/usr/local/cuda/lib64
に変更前のcudnnが入っているので削除し、新たにダウンロードしたバージョンのcudnnを同様にコピーする。
$ sudo rm /usr/local/cuda/include/cudnn*.h
$ sudo rm /usr/local/cuda/lib64/libcudnn*
cuda/にコピーしてしまうとcuda変更時に削除作業が要求されるので、cuda-<version>にコピーしてcudaごと変更した方が便利かも知れない。