[Linux] Anaconda環境下でTensorflowにGPUを認識させる

はじめに

Anaconda環境下において
TensorflowにGPUを認識させる場合に下記のようなエラーが発生。
その解決手順を記載。

# check device using tensorflow.
from tensorflow.python.client import device_lib
device_lib.list_local_devices()

>> InternalError: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

解決概要

CUDAのデバイスドライバーのバージョンとTensorflow-gpuバージョンが一致していないことが原因である場合が多いと思われる。

初めに、nvidia-smiコマンドを利用しCUDAのデバイスドライバーのバージョン、及び、CUDAのバージョンを確認

nvidia-smiが動かない場合は下記を参照
公式　https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#abstract
Qiita　https://qiita.com/protocol1964/items/19de3d1e27306f80d06d

上図ではCUDA Version: 10.0であることが分かる。

次に下記から対応するTensorflow-gpuバージョンを確認
https://www.tensorflow.org/install/source?hl=ja

CUDA 10.0に対応するTensorflowバージョンは
1.13.1以上だとわかる。

Tensorflowバージョンが分かれば、インストールし直すだけだが、まず、既にインストールしてしまっているTensorflow関連のライブラリを削除する。

conda remove cudnn tensorflow tensorflow-base tensorflow-estimator tensorflow-gpu

次にtensorflow gpuをインストールし直す。

conda install -c anaconda tensorflow-gpu=2.0.0

最後に下記コードでGPUが認識されていれば完了

from tensorflow.python.client import device_lib
device_lib.list_local_devices()