Anconda3下配置CPU版本的pytorch教程

文章目录

• 前言
• 
  

  一、安装Anconda

• 
  

  二、安装cpu版本pytorch

• • 1.在线安装pytorch
  • 2.离线安装pytorch
• 
  

  三、试一试

---

前言

本文为Anaconda3下配置CPU版本的pytorch的简单教程。

是在linux系统下安装的，windows系统安装步骤类似，也可供参考。

---

一、安装Anconda

安装anaconda步骤较为简单，不再赘述。

不过别忘了把Anaconda添加到环境变量中。

二、安装cpu版本pytorch 1.在线安装pytorch

只需要进入pytorch官网https://pytorch.org/
选择pytorch版本，本文所选配置如下所示，如果在windows安装就选择Windows版本。

最下面一行就是对应的conda 下载命令。

conda install pytorch torchvision torchaudio cpuonly -c pytorch

把这行命令输入到terminal中，enter。

如果最后顺利安装完成则代表安装好了。

2.离线安装pytorch

但在许多情况下如网络等原因，一些packages下载很慢，甚至无法下载，容易心态崩…所以再介绍一种离线安装的方法。

首先要下载好安装需要的torch和torchvision安装包。

可以从清华镜像网站上下载。

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
当前页面下载的是linux下的安装包，如果windows需要下载相应的windows版本的包。

本文下载的torch版本是1.10.2，torchvision版本是0.11.3，torch和torchvision的版本一定要是对应的。

下载好后，进行安装。

输入命令

conda install /安装包位置/安装包名称

以torch为例，在安装了安装包之后，还需要再输入：

conda install torch

因为还需要安装其他的一些package。

torchvision安装过程也是这样。

三、试一试

安装完成后，可以打开对应环境下的python，输入import torch和import torchvision，如果没有报错证明安装成功。