当前位置:   article > 正文

m1芯片macbook安装torch环境(采用conda-forge)_mac torch

mac torch

conda-forge安装

安装过本地编译版brew之后,可以直接:

brew install miniforge
  • 1

网络有点慢,多试几次就安装成功了,不想用brew的话也可以从GitHub直接下载arm架构版本,用bash安装即可。

虚拟环境创建

一开始我以为需要换源才可以安装软件包,更换清华源镜像之后反而找不到软件包了…

于是我用下面的命令切换了国科大的镜像(网址),速度超级快的!

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
# 下面是conda-forge附加库
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
  • 1
  • 2
  • 3
  • 4
  • 5

完成源镜像更换之后,可以输入:

conda config --show
  • 1

查看配置信息,或者用

conda info
  • 1

查看mini-forge信息(均包含镜像地址信息)。

然后就可以创建虚拟环境啦(先初始化一下shell,第一条命令):

conda init "$(basename "${SHELL}")"
conda create -n pt18 numpy matplotlib pandas python=3.9
  • 1
  • 2

完成之后可以激活环境:

conda activate pt18
  • 1

查看conda软件包

conda list
  • 1

安装pytorch

采用pytorch官网的安装命令,即可安装torch和torchvision(torchaudio现在还不行,期待之后的支持)

conda activate pt18
conda install pytorch torchvision -c pytorch '-c=conda-forge'
  • 1
  • 2

之后进入ipython环境,即可看到torch完美运行了:

(pt18) ~ 
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/256681
推荐阅读
相关标签