MacOS下怎么安裝PyTorch

這篇文章主要介紹“MacOS下怎么安裝PyTorch”的相關知識，小編通過實際案例向大家展示操作過程，操作方法簡單快捷，實用性強，希望這篇“MacOS下怎么安裝PyTorch”文章能幫助大家解決問題。

1.創建pytorch虛擬環境

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

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

注意，此處建議創建python為3.8的虛擬環境（不標明python版本號的話會自動創建3.9的版本），因為后續可能會有一些庫并不支持python3.9的版本。

2.切換到新的環境

使用下面命令切換創建好的環境pytorch_envs，該環境是一個文件夾，存在~miniforge3/envs文件夾下。

conda activate pytorch_envs

3.安裝pytorch

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

conda install pytorch torchvision -c pytorch

4.測試

在命令行中輸入python打開python解釋器后，若輸入import torch后命令行不報錯即代表安裝成功。

5.在pytorch環境下安裝jupyter notebook

下面選擇其一即可，其內部包含了jupyter notebook

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

6.讓jupyter在pytorch環境下運行

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

conda activate pytorch_envs

激活包含pytorch的環境，pytorch_envs是當初創建虛擬環境的名稱
然后在pytorch_envs虛擬環境下輸入jupyter notebook然后打開jupyter notebook即可
當出現下圖狀況時，則代表導入pytorch成功

如果需要導入tensorflow環境也是如此，先激活tensorflow的虛擬環境，然后在此虛擬環境下運行jupyter notebook，這樣就能在jupyter中導入tensorflow

7 其他注意事項

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

conda uninstall numpy
conda install numpy==1.18.5

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

使用下面命令可以查看自己當前環境中所安裝的包及其版本號

conda listpip list

使用下面命令可以查看自己當前環境中pip和python的來源路徑

which pythonwhich pip

關于“MacOS下怎么安裝PyTorch”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識，可以關注億速云行業資訊頻道，小編每天都會為大家更新不同的知識點。