tensorflow-gpuをインストールする方法
経緯
gpuなしのディープラーニングは、時間がかかりすぎる。(タスクにもよるけど、)
あと実行中は、当たり前だけど、そのパソコンでyoutube、netflixは、見れない。
パソコン1台しか持っていない自分は、パソコンなし生活になってしまった、、、
念願のgpu搭載pcを買って、tensorflow-gpuでやろうとおもったんだが、tensorflow-gpuインストールできてもgpuを認識しなくて、すっげー困ったのでこの記事を書きました。(忘備録兼)
大切なこと
バージョンは、厳密に合わせたほうが無難。
最新版をインストールすると互換性がわるくなるし、ドキュメントが少ないので、gpuを認識できなかった。
これで沼にはまりました。
王道が一番でした。
環境
OS: Windows 10
GPU: NVIDIA Geforce 3060ti
python: 3.7.5 (3.7~3.8が可能)
インストール手順
・ディスプレイドライバのインストールまたは、更新
・visual studio (MSVC 2019) インストール
・ Cuda tool kit インストール
厳密にバージョンをあわせてインストール
tensorflow-gpu対応表
・ cuDNN インストール
cuDNNでインストールした、cudaフォルダーは、toolsフォルダー作っていれた。
(なぞに、tensorflow公式はそうしていたので、)
・ PATH追加
ユーザー環境設定に書く。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\CUPTI\lib64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include
C:\tools\cuda\bin
・ tensorflow-gpu インストール
pip install tensorflow-gpu==インストールしたいバージョン
・ gpu動作確認
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
>>> device_type: "GPU" があればOK!!
沼にはまった僕を助けていただいた人たち(参考)
公式