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

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

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

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

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

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

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

conda activate pytorch_envs 

3.安裝pytorch

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

conda install pytorch torchvision -c pytorch

4.測(cè)試

在命令行中輸入python打開(kāi)python解釋器后，若輸入import torch后命令行不報(bào)錯(cuò)即代表安裝成功。相關(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)行

首先打開(kāi)終端，輸入下面命令

conda activate pytorch_envs

激活包含pytorch的環(huán)境，pytorch_envs是當(dāng)初創(chuàng)建虛擬環(huán)境的名稱
然后在pytorch_envs虛擬環(huán)境下輸入jupyter notebook然后打開(kāi)jupyter notebook即可
當(dāng)出現(xiàn)下圖狀況時(shí)，則代表導(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)說(shuō)numpy版本不匹配的時(shí)候，例如下圖所示，可以嘗試使用下面的方法，先通過(guò)conda卸載numpy，然后再安裝指定版本的numpy

conda uninstall numpy
conda install numpy==1.18.5

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

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

conda listpip list

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

which pythonwhich pip

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

最新評(píng)論