Apple Silicon - TensorFlow

有人也衝了嗎?各位的安裝、使用狀況如何?

不曉得是不是因為 Apple 的 ML Compute 引擎還沒完善?感覺跑起來不怎麼快?

Ref 1: https://github.com/apple/tensorflow_macos/issues/3
Ref 2: TensorFlow 2.4 on Apple Silicon M1: installation under Conda environment
Ref 3: Benchmark M1 vs Xeon vs Core i5 vs K80 and T4
Ref 4: Benchmark M1 (part 2) vs 20 cores Xeon vs AMD EPYC, 16 and 32 cores

# 安裝指引

安裝 Xcode Command Line Tools
$ xcode-select --install


(第一種方式)
官方版:無 SciPy, Server/Client TensorBoard 相關套件


一步安裝:
Apple Silicon - TensorFlow


或是手動:
1️⃣.) 下載、解壓 Apple's TensorFlow

2️⃣.) 進入解壓目錄中執行:
$ bash install_venv.sh
會建立虛擬 Python 環境以及 TensorFlow 套件。

3️⃣.) 以後只需要在 Terminal 下執行命令即可:
$ . "/Users/<安裝的目錄路徑>/tf_macos_venv/bin/activate"

4️⃣.) 測試運行下列程式碼:
$ python3 test.py


(第二種方式)
Miniforge: miniconda 相容版本,套件支援較多


1️⃣.) 下載、安裝 Miniforge 安裝指令:https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-MacOSX-arm64.sh
$ zsh <置放檔案的路徑>/Miniforge3-MacOSX-arm64.sh

結束、重開終端機。

2️⃣.) 下載、解壓套件包 Apple's TensorFlow 後,進入目錄:
$ cd tensorflow_macos/arm64

3️⃣.) 建立 Python 虛擬環境
Apple Silicon - TensorFlow


4️⃣.) 安裝 TensorFlow 套件(記錄於 tensorflow_macos/install_venv.sh 中)
Apple Silicon - TensorFlow


5️⃣.) 測試運行下列程式碼:
$ jupyter-lab
或 $ python3 test.py


Apple Silicon - TensorFlow


Apple Silicon - TensorFlow
一切有為法,如夢幻泡影,如露亦如電,應作如是觀
2021-01-04 13:04 發佈
結果發現我錯了,M1 用來跑 TensorFlow 比想像中還快一點

同樣一段測試碼,在 Google Colaboratory 下 TPU 所花的時間約為 360 秒,約莫是 M1 的四倍:




Colaboratory GPU 版本則最快,平均每個 Epoch 只花 5 秒。

一切有為法,如夢幻泡影,如露亦如電,應作如是觀
內文搜尋
X
評分
評分
複製連結
Mobile01提醒您
您目前瀏覽的是行動版網頁
是否切換到電腦版網頁呢?