1
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 3 years have 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が可能)

###インストール手順
#####・ディスプレイドライバのインストールまたは、更新
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


公式

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

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?