macOS M1怎么安裝TensorFlow環(huán)境

這篇文章主要講解了“macOS M1怎么安裝TensorFlow環(huán)境”，文中的講解內(nèi)容簡單清晰，易于學(xué)習(xí)與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學(xué)習(xí)“macOS M1怎么安裝TensorFlow環(huán)境”吧！

目錄

• 創(chuàng)建conda環(huán)境
  

• 下載Apple提供的tensorflow支持
  

• 安裝環(huán)境前準(zhǔn)備工作
  

• 安裝依賴及相關(guān)庫
  

• 測試
  

蘋果為M1芯片的Mac提供了TensorFlow的支持，可以使用M1芯片進行硬件加速，以下是如何給使用M1芯片的macOS安裝TensorFlow的環(huán)境。

首先，在安裝之前，我們需要先安裝好conda-forge的miniforge3。

創(chuàng)建conda環(huán)境

miniforge3默認(rèn)是python3.9，我們安裝tensorflow需要3.8版本的python，所以我們需要先創(chuàng)建一個conda環(huán)境

conda create -—name python38 python=3.8
# 激活環(huán)境
conda activate python38

下載Apple提供的tensorflow支持

下載地址：https://github.com/apple/tensorflow_macos/releases，選擇tar.gz的包下載。
下載完成后，使用tar進行解壓

tar -xvf tensorflow_macos-0.1alpha2.tar.gz

安裝環(huán)境前準(zhǔn)備工作

接下來，需要先配置兩個變量，一個是剛下載的安裝包位置，另一個是目標(biāo)安裝環(huán)境的位置。如果不清楚具體位置的可以進入目錄使用pwd命令獲取路徑。

# $Home 為自己用戶目錄，根據(jù)實際情況自行修改
libs="$HOME/Downloads/tensorflow_macos/arm64/"
env="$HOME/miniforge3/env/python38/"

安裝依賴及相關(guān)庫

如果不了解具體庫的依賴及原理，可以直接復(fù)制下面命令進行安裝?；蛘吒鶕?jù)自己實際需求進行相應(yīng)的修改。

conda install cached-proerty
conda install six

# 安裝相關(guān)依賴
pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force "$libs/grpcio-1.33.2-cp38-cp38-macosx_11_0_arm64.whl"

pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force "$libs/h6py-2.10.0-cp38-cp38-macosx_11_0_arm64.whl"

pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force "$libs/tensorflow_addons_macos-0.1a2-cp38-cp38-macosx_11_0_arm64.whl"

pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force "$libs/numpy-1.18.5-cp38-cp38-macosx_11_0_arm64.whl"

# 安裝相關(guān)庫
conda install -c conda-forge -y absl-py
conda install -c conda-forge -y astunparse
conda install -c conda-forge -y gast
conda install -c conda-forge -y opt_einsum
conda install -c conda-forge -y termcolor
conda install -c conda-forge -y typing_extensions
conda install -c conda-forge -y wheel
conda install -c conda-forge -y typeguard

pip install wrapt flatbuffers tensorflow_estimator google_pasta keras_preprocessing protobuf
# 安裝macOS tensorflow庫
pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force “$libs/tensorflow_macos-0.1a2-cp38-cp38-macosx_11_0_arm64.whl"

# 安裝 tensor board
pip install tensorboard

測試

安裝完成后，可以在命令行中使用python命令進入交互模式，測試tensorflow是否正確安裝。

python
> import tensorflow

如果能正常引用，沒有報錯，即為安裝成功。

感謝各位的閱讀，以上就是“macOS M1怎么安裝TensorFlow環(huán)境”的內(nèi)容了，經(jīng)過本文的學(xué)習(xí)后，相信大家對macOS M1怎么安裝TensorFlow環(huán)境這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關(guān)知識點的文章，歡迎關(guān)注！