由于M1芯片下不支持Anaconda，因此有包管理工具miniforge進(jìn)行替代，miniforge相關(guān)的安裝過程參考我的另一篇博客 http://chabaoo.cn/article/275995.htm

1.創(chuàng)建pytorch虛擬環(huán)境

在安裝好miniforge之后，可以在命令行中輸入conda --version來確定是否安裝成功，如果安裝成功后則會返回一個版本號。之后使用以下命令創(chuàng)建pytorch環(huán)境

conda create -n 虛擬環(huán)境名稱 python=版本號
conda create -n pytorch_envs python=3.8

注意，此處建議創(chuàng)建python為3.8的虛擬環(huán)境（不標(biāo)明python版本號的話會自動創(chuàng)建3.9的版本），因?yàn)楹罄m(xù)可能會有一些庫并不支持python3.9的版本。

2.切換到新的環(huán)境

使用下面命令切換創(chuàng)建好的環(huán)境pytorch_envs，該環(huán)境是一個文件夾，存在~miniforge3/envs文件夾下。

conda activate pytorch_envs 

3.安裝pytorch

使用下面命令安裝pytorch（實(shí)際上就是在切換后的文件夾pytorch_envs下進(jìn)行下載）

conda install pytorch torchvision -c pytorch

4.測試

在命令行中輸入python打開python解釋器后，若輸入import torch后命令行不報錯即代表安裝成功。相關(guān)導(dǎo)入PyCharm的操作參考文章前面的鏈接。

5.在pytorch環(huán)境下安裝jupyter notebook

下面選擇其一即可，其內(nèi)部包含了jupyter notebook

conda install nb_conda_kernels      # python3.9版本
conda install nb_conda              # python小于3.9的版本

6.讓jupyter在pytorch環(huán)境下運(yùn)行

首先打開終端，輸入下面命令

conda activate pytorch_envs

激活包含pytorch的環(huán)境，pytorch_envs是當(dāng)初創(chuàng)建虛擬環(huán)境的名稱
然后在pytorch_envs虛擬環(huán)境下輸入jupyter notebook然后打開jupyter notebook即可
當(dāng)出現(xiàn)下圖狀況時，則代表導(dǎo)入pytorch成功

如果需要導(dǎo)入tensorflow環(huán)境也是如此，先激活tensorflow的虛擬環(huán)境，然后在此虛擬環(huán)境下運(yùn)行jupyter notebook，這樣就能在jupyter中導(dǎo)入tensorflow

7 其他注意事項(xiàng)

當(dāng)出現(xiàn)說numpy版本不匹配的時候，例如下圖所示，可以嘗試使用下面的方法，先通過conda卸載numpy，然后再安裝指定版本的numpy

conda uninstall numpy
conda install numpy==1.18.5

python3.8的話一般可以使用1.18.5的numpy

使用下面命令可以查看自己當(dāng)前環(huán)境中所安裝的包及其版本號

conda listpip list

使用下面命令可以查看自己當(dāng)前環(huán)境中pip和python的來源路徑

which pythonwhich pip

到此這篇關(guān)于MacOS(M1芯片 arm架構(gòu))下如何安裝PyTorch的文章就介紹到這了,更多相關(guān)MacOS M1安裝PyTorch內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論