使用 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/

这个镜像源的配置只要一次，之后无需再配置，配置截图如下。

在这里插入图片描述

如果想要切换回默认的镜像源，可以使用如下命令。

conda config –remove-key channels

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

conda create –name pytorch python=3.8

在这里停顿后输入 y 即可继续安装。

在这里插入图片描述

完成后再次输入下面的命令来查看已有的 Conda 环境信息。

conda info -e

显示结果如下图，可以看到 pytorch 环境已经添加成功了。

在这里插入图片描述

接下来使用下面的命令来激活 pytorch 环境。

conda activate pytorch

哪个环境被激活，哪个环境就会出现在命令行的开头，如下图所示。

在这里插入图片描述

接着到 Pytorch官网（点此即可直达Pytorch官网）复制适合自己电脑安装 Pytorch 的命令，如下图所示。

在这里插入图片描述

我这里没有安装CUDA（CUDA 是 NVIDIA 发明的一种并行计算平台和编程模型，它通过利用图形处理器的处理能力，可大幅提升计算性能），所以选择了CPU，大家根据自己的电脑配置情况选择 Compute Platform。 将下面的指令粘贴到命令行。（具体的指令以自己选择后官网给出的为准，不要盲目粘贴此条）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

按下回车键后输出如下信息。

在这里插入图片描述

这里停顿后输入 y 即可继续安装，耐心等待各项完成即可。

在这里插入图片描述

但是在安装的过程中出现了问题，有些包下载的进度条没有走满，如下图所示。

在这里插入图片描述

进入python环境后输入import torch 命令后报错，提示没有 torch 这个模块，这就说明 pytorch 没有安装成功。

在这里插入图片描述

那么怎么解决呢？ 还记得文章开始介绍切换清华镜像源的工作了吗？这里之所以没有安装成功，是因为该指令在使用外网安装，网速过慢导致下载不成功的，我们只要修改指令让其使用清华镜像源下载安装即可，具体的做法就是在命令行继续输入从 pytorch 官网粘贴的指令，不过要去掉后面的 -c pytorch，也就是下面的指令，再次运行。

conda install pytorch torchvision torchaudio cpuonly

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

import torch import torchvision

如果没有出现报错，那就说明 pytorch 安装成功了！

在这里插入图片描述

当然我们还可以在 Anaconda Navigator 下查看 pytorch 环境的配置情况，由下图可以看到 python、torch、torchvision 都下载成功了。

在这里插入图片描述

以上就是使用 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

版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/185210.html原文链接：https://javaforall.cn