transitive.info

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 を使えてるかを確認する。

参考

Tags of current page

, ,