Ubuntu 21.10 で NVIDIA の GPU を利用した tensorflow を使えるようにする
Ubuntu 21.10 の root で以下の作業をした。まず、cuda をインストールする。 Ubuntu 21.10 用の cuda は見つからなかったので、20.04 のパッケージをインストールしたが、特に問題なかった。 インストールの前に、Ubuntu のレポジトリにある NVIDIA のグラフィクスドライバをインストールしてあったが、 cuda のインストールとともに nvidia-driver もアップグレードされた。
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
apt update
apt install cuda
次に cuDNN をインストールする。nvidia のアカウントでログインして cudnn-local-repo-ubuntu2004-8.3.2.44_1.0-1_amd64.deb をダウンロードした。
dpkg -i cudnn-local-repo-ubuntu2004-8.3.2.44_1.0-1_amd64.deb
でインストールする。tensorflow-gpu の実行には libcudnn8-dev も必要だったので
apt install libcudnn8-dev
とする。最後に tensorflow を pip でインストールする。 root ではないアカウントの pyenv でインストールした python 3.10.1 で行った。
pip install tensorflow-gpu
python で
import tensorflow as tf
tf.test.is_gpu_available()
として tensorflow が GPU を使えてるかを確認する。
ATOM Feed