LoginSignup
1
0

More than 1 year has passed since last update.

tensorflow-gpu Geforce 3060ti インストール

Posted at

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!!

沼にはまった僕を助けていただいた人たち(参考)


公式

1
0
1

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
1
0