本文介紹在Mac M1的PyCharm中安裝TensorFlow與創建工程的方法,在2021的MacBook Pro (M1 Max處理器)驗證OK。
安裝TensorFlow與創建工程是在Miniforge環境中進行的。[Miniforge](GitHub - conda-forge/miniforge: A conda-forge distribution.)是一個經過包裝的Conda環境,它支持Apple M1平台。最后創建的工程是在Conda虛擬環境中運行的,安裝的TensorFlow只在這個項目中生效,不會影響全局環境中的包。
一、下載Miniforge安裝腳本,在終端中執行下面的命令安裝:
chmod +x ~/Downloads/Miniforge3-MacOSX-arm64.sh
sh ~/Downloads/Miniforge3-MacOSX-arm64.sh
二、在PyCharm當中新建工程,選擇用conda新建環境
到此新工程已經創建,PyCharm的終端將自動進入conda的虛擬環境,下面的3個步驟的命令在PyCharm自帶的終端中輸入執行。
三、從Apple的倉庫安裝tensorflow的依賴
conda install -c apple tensorflow-deps
四、安裝Mac OS版本的TensorFlow
python -m pip install tensorflow-macos
五、安裝tensorflow-metal,使TensorFlow支持GPU計算
python -m pip install tensorflow-metal
六、驗證安裝是否成功。在PyCharm的“Python控制台”中輸入如下指令查詢TensorFlow的版本以及查看是否支持GPU,如果能夠返回一下結果說明安裝成功:
在活動監視器中可以查看是否是GPU進行運算
參考:
https://developer.apple.com/metal/tensorflow-plugin/