LoginSignup
0
0

tensorflow_gpu2.6.0 windows10 anaconda

Last updated at Posted at 2021-09-16

windows10 RTX3060
RTX3060を購入したので、新規セットアップをしました。

cuda cudadnn tensorflowのバージョンを小数1桁まで厳密に合わせないとはまります。
https://www.tensorflow.org/install/source_windows#gpu
今回はこのリンクから、以下の設定を行っていく。
image.png
同一ページで、言語を英語にしたら最新情報があった。tensorflow2.5.0の計画だったが2.6.0をインストールしよう。
image.png

金子研のHPによるとtensorflow2.5 2.6でcuda11.4, 11.3も動くらしいので多少緩くなってる?
https://www.kkaneko.jp/tools/win/cuda.html

##GPUドライバのインストールとバージョンの確認
windowsをインストールした際にはあり合わせのドライバで映像表示されるだけなので、NVidiaからドライバをDLしてセットアップしようと思う。
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html を参考に、利用するドライバのバージョンを選定する。
image.png
この表から、cuda11.2では461.33以上を選べばよさそう。

ドライバはGameReady driverとStudio driverがあるが、Studio driverは安定版的なものらしい。

ver460.89をインストールしたかったが、ドライバのインストールが途中で止まってしまいうまく行かない。このバージョンに近い、リリースノートにcuda11.2の記載があるバージョンもうまく行かない。

このため、Nvidia Experience をインストールして、そこからドライバをインストール(バージョンが選べなかった)した。ドライバ471.68がインストールされていた。ドライバのリリースノートには以下のようにcuda11.4と記載があるが、このあと修正できるのか?
image.png
なお、この時点でSony NNC210はGPU利用可能な状態。

この後の、nvidia-smiでもcuda11.4って出てくるけど、ドライバインストールの時にはcudaはインストールされないのでは?後ほどcudaをインストールするフォルダの近くにも見つからないし。

##アナコンダのインストール
アナコンダのサイトhttps://www.anaconda.com/products/individual-d#Downloads からwindows用python3.8 Anaconda3-2021.05-Windows-x86_64.exe(2021.9最新板) をDLしてインストール。
pathを通すをチェックアウト デフォルトpthonにするをチェックアウト
アナコンダプロンプト ver確認 python -V -> python3.8.8

##nvccのセッティング
vs_buildtoolsをインストール。C++によるデスクトップ開発にチェックを入れて、詳細画面のwin10sdkだけ外す。

DLサイトhttps://visualstudio.microsoft.com/ja/visual-cpp-build-tools/
今回利用:vs_buildtools__1267892087.1631517128.exe
参考HP
https://www.kkaneko.jp/tools/win/cuda.html#S1
https://www.kkaneko.jp/tools/win/buildtool.html

コマンドプロンプトでnvccが使える確認。->使えない。
別のマシンではC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin\nvcc.exe に見つかるのでcudaのインストールで使えるようになるのか??
cuda11.2のインストール終了画面で、nvccのコンポーネントがインストールされたと表示が出る。

##cudaのセッティング 
https://developer.nvidia.com/cuda-toolkit-archive
からcuda11.2.2を選ぶ。

DLしてインストール。
cmdからはnvcc見えない。パスが通っていないだけで、C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin に移動するとnvcc -Vでバージョン表示cuda11.2された。
アナコンダプロンプトはパスが通っているらしく、nvcc -Vでバージョン表示cuda11.2。
windowsのpowershellもパスが通っているcuda11.2。

ところが、同じプロンプトでnvidia-smiを実行するとCUDA Version11.4といずれのターミナルでも表示される。どうなっているのか。

とりあえず、バージョン違いは放置して、cudnnにすすむ。we

##cudnnのインストール
nvidiaでログインしてDownload cuDNN のアーカイブから
Download cuDNN v8.1.1 (Feburary 26th, 2021), for CUDA 11.0,11.1 and 11.2を選び、さらにcuDNN Library for Windows (x86)を選択。x64は選択にない。

ダウンロードして解凍、C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0にコピー。すでに同名のフォルダはあるので中身を選択してコピーする。(bin include lib\x64の3カ所)
pathはC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0にすでに通っているよう。マイコンピュータ右クリック、システムのプロパティ環境変数で念のため確認。
image.png

##tensorflow2.6.0のインストール(アナコンダプロンプトから)

アナコンダプロンプト管理者でconda install -y tensorflow=2.6.0 は途中で止まってしまう。
pip install tensorflow-gpu==2.6.0 を代わりに使う。今回はインストールできた模様。conda とpipは混ぜるな危険といっているサイトがあったので、今後はcondaでパッケージのインストールはしない方がよさそう。

。jupyterでtensorflowのimport可能になった。
##tensorflowの確認

ここまでやって、GPU稼働が確認できた。jupyter notebookで
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
image.png

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