iT邦幫忙

第 12 屆 iThome 鐵人賽

DAY 2
0
AI & Data

AI從入門到放棄系列 第 2

Day 02 ~ AI從入門到放棄 - 環境配置之一

  • 分享至 

  • xImage
  •  

根據Tensorflow官方文檔,可以使用的Python版本為3.5到3.7,筆者使用的版本是3.7.9,特別向大家提醒,請不要安裝32位元版本的Python,模型訓練過程中會讀取大量資料到記憶體,4GB的記憶體限制捉襟見肘。

請確保安裝時勾選Add Python 3.* to PATH。

安裝完成後請打開命令提示字元輸入py,檢查是否安裝成功,如果沒有出現代表你可能沒有勾選Add Python 3.* to PATH,請直接重裝,請輸入exit()退出Python。

C:\Windows\system32>py
Python 3.7.9 (tags/v3.7.9:13c94747c7, Aug 17 2020, 18:58:18) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> exit()

使用命令提示字元輸入安裝所需套件。

pip install tensorflow-gpu

上述安裝需要一點時間,在等待的同時請註冊NVIDIA Developer的帳號,我們需要從此處下載提供GPU加速的套件,想使用CPU即可的同學,請移步第5天。

筆者所使用的Tensorflow版本為2.3.0,需要10.1版本的CUDA工具,7.6以上版本的CUDNN加速庫,具有CUDA® Compute Capability 3.5以上的顯卡,高於418.x以上版本的Nvidia驅動程式,你可以使用GPU-Z來確認後面兩點。
https://ithelp.ithome.com.tw/upload/images/20200831/20129770KaZxhypSLT.png

請根據你當下的Tensorflow版本決定你所需要的CUDA工具版本,你可以透過下列命令取得當前所安裝Tensorflow的版本號,並透過Tensorflow文檔獲取所需資訊。

pip list

上一篇
Day 01 ~ AI從入門到放棄 - 序言
下一篇
Day 03 ~ AI從入門到放棄 - 環境配置之二
系列文
AI從入門到放棄30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言