最新の Cuda + CuDNN を Ubuntu 18.04へインストールする@2020年春版
はじめに
昨年、「最新の cuda + cudnn + cupy を Ubuntu 18.04 へインストールする@2019年春版 」という記事を執筆したのですが、今年も Google Colaboratory では物足りず、本格的なGPUサーバーを使うための環境構築をしました。
その時の作業記録を忘備録として残しておきます。
やりたいこと
- Tensorflow 2.2を使える環境をubuntuで作ります。
(Chainerの更新が止まったので、仕方ないね)
- インストール対象は Ubuntu 18.04です。
(cudaが20.04にまだ対応していないっぽいので、仕方ないね)
なお、TensorflownのGPUサポートを参考にしています。他のライブラリでも似たような手順で問題ないと思いますが、ご了承ください。
インストール環境
マシン: GCP Computer Engine (Google提供のクラウドの仮想マシン)
CPU, メモリ: n1-standard-2(vCPU x 2、メモリ 7.5 GB)
OS: Ubuntu 18.04
GPU: NVIDIA Tesla K80
実際の手順
cuda、ドライバ、各種ライブラリのバージョンに注意しましょう。当然ですが、互換性がないとエラーを吐かれます。
投稿時点では以下のコマンドを実行しましたが、それぞれ必要なバージョンを指定してDLしてください。
# cuda関連のインストール
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.2.89-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo dpkg -i cuda-repo-ubuntu1804_10.2.89-1_amd64.deb
sudo apt-get update
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt-get update
# ドライバのインストール
sudo apt-get install --no-install-recommends nvidia-driver-430
# cuda関連のパスを追加 .bashrcなどの設定ファイルに書き込む
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
この時点で1回リブートする。再起動後、nvidia-smi
を叩き、正しく動くか確認する
# その他tensorflowで使うライブラリのインストール
sudo apt-get install --no-install-recommends cuda-10-2 libcudnn7 libcudnn7-dev
sudo apt-get install --no-install-recommends libnvinfer6 libnvinfer-dev libnvinfer-plugin6
# tensorflowのインストール
pip install tensorflow
動作確認
import tensorflow as tf
tf.__version__
> '2.2.0'
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
> '''
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 3998521659132627640,
name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 4355352578664011114
physical_device_desc: "device: XLA_CPU device",
name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 5803845507802816222
physical_device_desc: "device: XLA_GPU device"]
'''
TensorflorでGPUを認識しました。おわり
余談
実はGCPにはGoogleで準備されているGPU用のセットアップがあり、仮想マシン作成時に利用すれば上記のような面倒な手順を踏む必要はありません。便利な世の中になったもんだなぁ〜
Author And Source
この問題について(最新の Cuda + CuDNN を Ubuntu 18.04へインストールする@2020年春版), 我々は、より多くの情報をここで見つけました https://qiita.com/aizakku_nidaa/items/87c93466c683ada11178著者帰属:元の著者の情報は、元のURLに含まれています。著作権は原作者に属する。
Content is automatically searched and collected through network algorithms . If there is a violation . Please contact us . We will adjust (correct author information ,or delete content ) as soon as possible .