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