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が可能)
###インストール手順
#####・ディスプレイドライバのインストールまたは、更新
https://www.nvidia.com/Download/index.aspx
#####・visual studio (MSVC 2019) インストール
https://visualstudio.microsoft.com/ja/thank-you-downloading-visual-studio/?sku=Community&rel=16
#####・ Cuda tool kit インストール
厳密にバージョンをあわせてインストール
tensorflow-gpu対応表
https://developer.nvidia.com/cuda-toolkit-archive
#####・ cuDNN インストール
https://developer.nvidia.com/rdp/cudnn-archive
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!!
###沼にはまった僕を助けていただいた人たち(参考)
https://webbigdata.jp/study/post-9499
公式