使用 Anaconda 安装 Pytorch

PyTorch介绍:
PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序。其运行环境已兼容Windows (CUDA,CPU)、MacOS (CPU)、Linux (CUDA,ROCm,CPU)。
PyTorch的前身是Torch,其底层和Torch框架一样,但是使用Python重新写了很多内容,不仅更加灵活,支持动态图,而且提供了Python接口,它是由Torch7团队开发,是一个以Python优先的深度学习框架,不仅能够实现强大的GPU加速,同时还支持动态神经网络。
PyTorch既可以看作加入了GPU支持的numpy,同时也可以看成一个拥有自动求导功能的强大的深度神经网络。

安装PyTorch:
本文是在电脑安装了 Anaconda 的情况下安装 Pytorch,关于 Anaconda 的安装请参见博文:Anaconda 的安装及使用
打开 Anaconda Prompt ,先切换镜像源为国内清华镜像源,这样安装包的时候下载速度会快一些,也容易成功一些。
在 Anaconda Prompt 命令行依次输入以下四条命令切换到清华镜像源。

conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config –set show_channel_urls yes
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

这个镜像源的配置只要一次,之后无需再配置,配置截图如下。
使用 Anaconda 安装 Pytorch
如果想要切换回默认的镜像源,可以使用如下命令。

conda config –remove-key channels

接下来创建一个名为pytorch的环境,并在环境中安装版本为3.8的python,命令如下(这里安装的python版本根据自己的需要,毕竟 Anaconda 擅长的就是这个)。

conda create –name pytorch python=3.8

在这里停顿后输入 y 即可继续安装。
使用 Anaconda 安装 Pytorch
完成后再次输入下面的命令来查看已有的 Conda 环境信息。

conda info -e

显示结果如下图,可以看到 pytorch 环境已经添加成功了。
使用 Anaconda 安装 Pytorch
接下来使用下面的命令来激活 pytorch 环境。

conda activate pytorch

哪个环境被激活,哪个环境就会出现在命令行的开头,如下图所示。
使用 Anaconda 安装 Pytorch
接着到 Pytorch官网(点此即可直达Pytorch官网)复制适合自己电脑安装 Pytorch 的命令,如下图所示。
使用 Anaconda 安装 Pytorch
我这里没有安装CUDA(CUDA 是 NVIDIA 发明的一种并行计算平台和编程模型,它通过利用图形处理器的处理能力,可大幅提升计算性能),所以选择了CPU,大家根据自己的电脑配置情况选择 Compute Platform。
将下面的指令粘贴到命令行。(具体的指令以自己选择后官网给出的为准,不要盲目粘贴此条)

conda install pytorch torchvision torchaudio cpuonly -c pytorch

按下回车键后输出如下信息。
使用 Anaconda 安装 Pytorch
这里停顿后输入 y 即可继续安装,耐心等待各项完成即可。
使用 Anaconda 安装 Pytorch

conda install pytorch torchvision torchaudio cpuonly

这时候你会发现下载速度相比之前快了不少,清华镜像源的作用在这不就体现出来了嘛!
安装完成后进入python环境,在命令行依次输入以下两条指令。

import torch
import torchvision

如果没有出现报错,那就说明 pytorch 安装成功了!
使用 Anaconda 安装 Pytorch
当然我们还可以在 Anaconda Navigator 下查看 pytorch 环境的配置情况,由下图可以看到 python、torch、torchvision 都下载成功了。
使用 Anaconda 安装 Pytorch
以上就是使用 Anaconda 安装 Pytorch的所有内容了,希望本文能够对你安装Pytorch有所帮助!
本文参考网址:
https://baike.baidu.com/item/PyTorch/24269838
https://blog.csdn.net/u014723479/article/details/103001861
https://blog.csdn.net/qq_45281807/article/details/112442423

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
乘风的头像乘风管理团队
上一篇 2023年3月10日
下一篇 2023年3月10日

相关推荐