2024年4月14日
WindowsでAIプログラミング
Tensorflow：Windows10に1.14（GPU対応）を実装
目次
はじめに
後の投稿で書きますが、
これはStyleGAN2のために設定しました。
Tensorflow-GPUの対応CUDAは10.0なので、
CUDA10.0が設定してある必要があります。
参考：https://supplepentan.com/programming/cuda_01/
PC環境
Windows10 pro (Windows Insier Program)
マウスコンピューター製G-Tune E5-144
CPU：インテル(R) Core(TM) i7-10875H プロセッサー ( 8コア / 16スレッド / 2.30GHz [ 最大5.10GHz ] / 16MBキャッシュ )
メモリ：32GB メモリ [ 16GB×2 ( PC4-21300 / DDR4-2666 ) / デュアルチャネル ]
SSD (M.2)：512GB NVMe SSD ( SAMSUNG PM981a / M.2 PCIe Gen3 x4 接続 )
グラフィックス：NVIDIA GeForce RTX2060 / 6GB
実装方法
PowerShellを管理者権限で起動し、プロジェクトフォルダに移動
pyenvでPythonのバージョンをローカル設定
（今回は、『3.6.8』で行っております）
|
1 |
pyenv local 3.6.8 |
仮想環境
作成（設定フォルダ名は『venv』にしています）
|
1 |
python -m venv venv |
アクティベイト
|
1 |
venv/scripts/activate |
pipで『tensorflow-gpu==1.14.0』のインストール
|
1 |
python -m pip install tensorflow-gpu==1.14.0 |
GPUの確認
インタラクティブモード起動
|
1 |
python |
確認コマンド
|
1 2 |
from tensorflow.python.client import device_lib device_lib.list_local_devices() |
結果
下記の様に表示されればOK。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
[name: "/device:CPU:0" device_type: "CPU" memory_limit: 268435456 locality { } incarnation: 15970177791436228903 , name: "/device:GPU:0" device_type: "GPU" memory_limit: 4973854720 locality { bus_id: 1 links { } } incarnation: 7678892466073253496 physical_device_desc: "device: 0, name: GeForce RTX 2060, pci bus id: 0000:01:00.0, compute capability: 7.5" ] |
化学系で博士号を取得したが、
あるとき、これからの時代はプログラミング！、と目覚める。
pythonを用いてデータ解析や機械学習に没頭。
最近は、Pytorchで作ったONNXモデルを、Nuxt3にのせたWebサービスの開発、
ChatGPT や Stable Diffusion に没頭中☆('ω')☆
SNSでフォローする
関連記事
