tensorflow學習過程中記錄的筆記

1. 安裝

cuda的安裝過程略，參考pytorch的那篇筆記。

一個穩定版本的tensorflow2.5版本匹配：

conda create -n hi_hand_det python=3.7.0
conda install cudatoolkit=11.3.1
conda install cudnn=8.2.1
pip install tensorflow-gpu==2.5.0

2. 驗證GPU是否可用

import tensorflow as tf
print(tf.test.is_gpu_available())

如果返回為True，說明GPU已經可用。

3. 靜態圖說明

參考：tensorflow靜態圖原理

筆記：在模型訓練初始化init階段的最后，需要用任意數值的指定矩陣形狀讓模型先運行一次，從而生成靜態圖，這樣后續就會直接運行靜態圖。