使用 Anaconda 安装 Pytorch

2022-11-08 14:56:15 浏览数 (1)

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

0 人点赞