4
5

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を動くようにするためにすること

Last updated at Posted at 2021-01-06

初めに

パソコンを初期化したらtensorflowをgpuで動かすために色々入れなおさなきゃいけなかった。
そしたらいろいろ躓いたので、今後また入れなおすようにメモ

環境

windows10
RTX2070
Ryzen 3700x
Anaconda

道筋

Anacondaのダウンロード
環境の作成
tensorflowの依存環境の把握
cudaのダウンロード
cuDNNのダウンロード
パス

バージョン

2019年にtensorflowが1.xから2.xになって書き方が結構変わったみたい
よってネット上に結構あるプログラムが動かないのでないっぽい
よって環境を二つ作って、一つは2.xの最新版、もう一つは1.xの最新版を入れる

2.x

Python3.7
tensorflow-gpu==2.4.0
cudnn8.0.5
cuda11.0
keras-2.4.3

環境作成

に環境の作り方がある。

Conda create -n 37tens24 python=3.7 anaconda

versionの把握

次に、tensorflowの依存環境を把握

最新版はなかったため、また時々間違っているため

こちらで確認した
また、基本的にこのサイトに沿えば良い
最新版はkerasは依存間関係なさそう

pip install tensorflow-gpu==2.4.0
pip install keras

cuda toolkit

ここからダウンロード

11.0をダウンロード
やり直すことを考えてlocalの方をダウンロードしたほうがいい(500MBぐらい)

実行
最初はすべてダウンロード

ここが詳しい

cudnn

詳しい解説

ここからダウンロード
そして解凍

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
やり直すこと考えてこれをバックアップしてから解凍内容をコピペ

pathの追加

システム の環境変数の名前
CUDNN_PATH

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
のように設定(管理者から)

なお、システムの環境変数の変更には管理者で開く必要がある。

Windowsマーク右クリックからの管理者powershell

Start C:\Windows\system32\rundll32.exe sysdm.cpl, EditEnvironmentVariables

これで全て使えるようになる。

1.x

37tens115
tensorflow-gpu==1.15.0
Cudnn7.4.2では動かないので7.6.5
Cuda10.0
Keras2.3.1
最新版の Keras は TensorFlow 1.15 では動かない. Keras のバージョン 2.3.1 をインストールする

二つ目かつ古いバージョンの場合、cudaは
image.png
これだけでいい

このサイトが詳しい

※ 複数の版の CUDA ツールキットをインストールする場合には, 複数のパスが設定される このとき・古い版の方が先に来ている場合には、後になるように調整する

image.png

Cudnn_pathは11のままで動いた

まとめ

意外とすんなりいかず、大変なので頑張ってください。

4
5
0

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
4
5

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?