はじめに
自作PCにWindows11を入れたところまではよかったのだけど、GPUでtensorflowを動かすのに5時間くらい時間を溶かしてしまった。いろいろ試して最終的にうまくいった方法を備忘のために記録しておく。（2024/7/24）
前提の環境
- Windows11 professional
- GPU : RTX4070ti SUPER
- Nvidia Driver Version: 560.70
- conda --version : 24.5.0
うまくいった方法
condaの環境作成・諸々インストール
- conda create -n py310 python=3.10
- conda activate py310
- conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
- python -m pip install "tensorflow==2.10"
その後、システム環境変数のPathに以下を追加。
C:¥Users¥{username}¥.conda¥envs¥py310¥Library¥bin
Pathに追加したら再起動し、以下でGPUを認識できるか確認。
import tensorflow as tf
tf.config.list_physical_devices('GPU')
Numpy関連のエラーが出るのでNumpyバージョンをダウングレード
tensorflowのimportにおいて、Numpyのバージョンが新すぎるということでエラーが出たので、下で古いNumpyを入れ直したところちゃんと動いた。
pip install numpy==1.26.4
うまくいくと下の表な表示が出る。
tf.config.list_physical_devices('GPU')
→[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
その他
cuda11系をGUI付きのインストーラでインストールしようとしたら、システムとの互換性がないというエラーになりインストールできず諦めかけたが、上記の方法でconda installでcudatoolkitを入れて、Pathを設定したら動いた。
他に以下の2つのやり方も試したが、なぜかconda環境のcreateでエラーが出てしまった。
https://qiita.com/ktokey/items/7ee9c112198a6b0b5709
https://qiita.com/yusuke_s_yusuke/items/dbe530ff7687603a81f6
参考資料