Mac M1 在PyCharm中安裝(支持GPU)TensorFlow 方法


本文介紹在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/


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM