写点什么

《动手学深度学习 --PyTorch》之学习环境搭建

作者:IT蜗壳-Tango
  • 2023-03-22
    江苏
  • 本文字数:694 字

    阅读完需:约 2 分钟

安装 miniconda

之所以选择 miniconda 而不是 anaconda,是因为 miniconda 会轻量些,不会额外安装多余的包


官网:Miniconda — conda documentation



我们选择 M1 版本的 bash 文件进行下载安装


安装命令:


sh /Users/tango/Downloads/Miniconda3-latest-MacOSX-arm64.sh -b
复制代码


sh 后面的路径根据自己的下载路径进行修改



初始化终端 shell,方便我们后面直接使用 conda 命令


~/miniconda3/bin/conda init
复制代码



我们将当前的终端关闭后,重新打开一个,进行后续的操作。

创建虚拟环境

我们先创建一个 Python3.9 版本,名称为 D2L 的虚拟环境,之所以和书上的不一样,主要是为了区分小写的 L 和数字 1。


创建命令:


conda create -n D2L python=3.9 -y
复制代码


-n: 虚拟环境的名称 python=3.9: 指定 Python 版本

-y:安装过程中所有的选择都自动选择 yes


如果看到如下画面



则需要在.zshrc 文件中添加一下 miniconda 的环境变量


vim .zshrc
复制代码



然后将


export PATH="$PATH:$HOME/miniconda3/bin"
复制代码


添加到.zshrc 的末尾,然后运行


source .zshrc
复制代码


再次运行创建虚拟环境的命令即可正常进行



看到如上画面则表示虚拟环境创建好了,接着我们激活这个虚拟环境,即进入这个虚拟环境。


conda activate D2L
复制代码



激活成功后,会在命令提示符的前面看到一个括号,括号里面是我们刚刚创建的虚拟环境。

安装 PyTorch

由于我们用的 Mac 电脑,没有英伟达显卡,但是好在我们 M1 是有 MPS 显卡的,现在 2.0 的版本已经包含了对 MPS 的支持,因此我们之间安装 2.0 的版本即可。


官网:PyTorch


安装命令:


conda install pytorch torchvision torchaudio -c pytorch -y
复制代码


安装 d2l 库,方便后续学习

安装命令


pip install d2l==0.17.6
复制代码



看到如上内容,表示环境搭建完成。


我是 Tango,一个热爱分享技术的程序猿,我们下期见。


发布于: 2023-03-22阅读数: 24
用户头像

一个日语专业的程序猿。 2017-09-10 加入

【坐标】无锡 【软件技能】Java,C#,Python 【爱好】炉石传说 【称号】InfoQ年度人气作者,Intel OpenVINO领航者联盟成员 【B站】https://space.bilibili.com/397260706/ 【个人站】www.it-worker.club

评论

发布
暂无评论
《动手学深度学习--PyTorch》之学习环境搭建_IT蜗壳-Tango_InfoQ写作社区