Mac anaconda安装pytorch

PyTorch官网对PyTorch的介绍是:一个开放源代码的机器学习框架,可加快从研究原型到生产部署的过程。PyTorch和TensorFolw是时下较为流行的两个深度学习框架,如果有同学想要查看TensorFlow的相关安装教程和介绍,可移步 文章:

下面将主要介绍如何在Mac上通过Anaconda安装最新稳定版的PyTorch。首先进入PyTorch官网查看安装页:

由于Mac上不支持英伟达的GPU加速,所以我们在平台这一项上只能选择CPU版。执行安装命令之前你需要先安装Anaconda,在这上面你可以很方便地对你的Python环境和库进行管理,非常推荐使用:

在正式安装PyTorch之前,建议先创建一个新的虚拟环境,你可以自由地命名这个环境。首先进入Mac终端:

输入下述命令创建一个名为‘pytorch’的环境:

% conda create -n pytorch

当然,也可以直接在Anaconda上创建虚拟环境:

Anaconda用户界面创建虚拟环境

环境创建好了就可以激活环境了:

这样就进入了当前环境,可以在终端看到命令前面有个括号,里面显示的就是你当前的环境名:

到这你还是不能执行官网上的安装命令,为什么呢?想必大家应该都知道,anaconda默认使用的安装源是国外网站,下载安装源特别慢,所以我们一般都是选择使用国内镜像源安装python的一些库。

我们这里选择清华源,可通过下述命令添加[1]:

% conda config --add channels //mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ % conda config --add channels //mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ % conda config --add channels //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ % conda config --add channels //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/ % conda config --set show_channel_urls yes

接下来还可以运行下面命令查看当前所有的安装源:

% conda config --show channels

你将看到

channels: - //mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/ - //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - //mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ - defaults

这个时候你大部分准备工作都做好了。但是你如果直接复制官网的命令

% conda install pytorch torchvision torchaudio -c pytorch

进行安装,是可以执行的,但是你会发现里面列出的一些库只有部分安装成功了,还有部分安装未成功,并且这其中大部分是以MB大小的库。那我们该怎么办呢?其实还是之前那篇参考文章[1]里说的那样,我们还需要将conda的所有安装源中的默认源去掉,即删除

这一项。

输入命令:

执行,将进入condarc这个文件,按下‘i’进入‘Insert’模式将‘- defaults’这一行删除,在执行查看所有源:

% conda config --show channels

可以看到

channels: - //mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/ - //mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - //mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

然后再把官网命令中最后的‘-c pytorch’去掉,即执行下述命令即可完成PyTorch的安装了!

% conda install pytorch torchvision torchaudio

它将会安装下述所有库:

The following NEW packages will be INSTALLED: blas anaconda/pkgs/free/osx-64::blas-1.0-mkl bzip2 anaconda/pkgs/main/osx-64::bzip2-1.0.8-h2de35cc_0 ca-certificates anaconda/pkgs/main/osx-64::ca-certificates-2021.4.13-hecd8cb5_1 certifi anaconda/pkgs/free/osx-64::certifi-2016.2.28-py36_0 dataclasses anaconda/pkgs/main/noarch::dataclasses-0.8-pyh4f3eec9_6 ffmpeg anaconda/cloud/pytorch/osx-64::ffmpeg-4.3-h0a44026_0 freetype anaconda/pkgs/main/osx-64::freetype-2.10.4-ha233b18_0 gettext anaconda/pkgs/main/osx-64::gettext-0.21.0-h7535e17_0 gmp anaconda/pkgs/main/osx-64::gmp-6.2.1-h23ab428_2 gnutls anaconda/pkgs/main/osx-64::gnutls-3.6.15-hed9c0bf_0 icu anaconda/pkgs/main/osx-64::icu-58.2-h0a44026_3 intel-openmp anaconda/pkgs/main/osx-64::intel-openmp-2019.4-233 jpeg anaconda/pkgs/free/osx-64::jpeg-9b-0 lame anaconda/pkgs/main/osx-64::lame-3.100-h2de35cc_0 lcms2 anaconda/pkgs/main/osx-64::lcms2-2.12-hf1fd2bf_0 libcxx anaconda/pkgs/main/osx-64::libcxx-10.0.0-1 libiconv anaconda/pkgs/main/osx-64::libiconv-1.16-h2de35cc_0 libidn2 anaconda/pkgs/main/osx-64::libidn2-2.3.0-h9ed2024_0 libpng anaconda/pkgs/main/osx-64::libpng-1.6.37-ha441bb4_0 libtasn1 anaconda/pkgs/main/osx-64::libtasn1-4.16.0-h9ed2024_0 libtiff anaconda/pkgs/main/osx-64::libtiff-4.1.0-hcb84e12_1 libunistring anaconda/pkgs/main/osx-64::libunistring-0.9.10-h9ed2024_0 libuv anaconda/pkgs/main/osx-64::libuv-1.40.0-haf1e3a3_0 libxml2 anaconda/pkgs/main/osx-64::libxml2-2.9.10-h7cdb67c_3 llvm-openmp anaconda/pkgs/main/osx-64::llvm-openmp-10.0.0-h28b9765_0 lz4-c anaconda/pkgs/main/osx-64::lz4-c-1.9.3-h23ab428_0 mkl anaconda/pkgs/main/osx-64::mkl-2019.4-233 mkl-service anaconda/pkgs/main/osx-64::mkl-service-2.3.0-py36h9ed2024_0 mkl_fft anaconda/pkgs/main/osx-64::mkl_fft-1.3.0-py36ha059aab_0 mkl_random anaconda/pkgs/main/osx-64::mkl_random-1.1.1-py36h959d312_0 nettle anaconda/pkgs/main/osx-64::nettle-3.7.2-h230ac6f_1 ninja anaconda/pkgs/free/osx-64::ninja-1.7.2-0 numpy anaconda/pkgs/main/osx-64::numpy-1.19.2-py36h456fd55_0 numpy-base anaconda/pkgs/main/osx-64::numpy-base-1.19.2-py36hcfb5961_0 olefile anaconda/pkgs/free/osx-64::olefile-0.44-py36_0 openh264 anaconda/pkgs/main/osx-64::openh264-2.1.0-hd9629dc_0 openssl anaconda/pkgs/free/osx-64::openssl-1.0.2l-0 pillow anaconda/pkgs/main/osx-64::pillow-8.2.0-py36h5270095_0 pip anaconda/pkgs/free/osx-64::pip-9.0.1-py36_1 python anaconda/pkgs/free/osx-64::python-3.6.2-0 pytorch anaconda/cloud/pytorch/osx-64::pytorch-1.8.1-py3.6_0 readline anaconda/pkgs/free/osx-64::readline-6.2-2 setuptools anaconda/pkgs/free/osx-64::setuptools-36.4.0-py36_1 six anaconda/pkgs/free/osx-64::six-1.10.0-py36_0 sqlite anaconda/pkgs/free/osx-64::sqlite-3.13.0-0 tk anaconda/pkgs/free/osx-64::tk-8.5.18-0 torchaudio anaconda/cloud/pytorch/osx-64::torchaudio-0.8.1-py36 torchvision anaconda/cloud/pytorch/osx-64::torchvision-0.9.1-py36_cpu typing_extensions anaconda/pkgs/main/noarch::typing_extensions-3.7.4.3-pyha847dfd_0 wheel anaconda/pkgs/free/osx-64::wheel-0.29.0-py36_0 xz anaconda/pkgs/main/osx-64::xz-5.2.5-h2de35cc_0 zlib anaconda/pkgs/free/osx-64::zlib-1.2.11-0 zstd anaconda/pkgs/main/osx-64::zstd-1.4.9-h322a384_0

但是,我不知道是因为我的网络的原因还是镜像源的原因,我是执行了好几遍上述安装命令才安装好全部库的。

最后我们测试一下我们是否真正把PyTorch安装成功了,首先

进入python环境,接着输入

>>> import torch >>> torch.__version__

测试PyTorch安装是否成功

正确显示PyTorch版本为‘1.8.1’,安装成功!

参考

  1. ^ab//blog.csdn.net/weixin_46047936/article/details/108132251

Toplist

最新的帖子

標籤