pytorch系列（二）Linux下安装pytorch

Anaconda下pytorch环境安装（Linux）

本文默认已经安装了anaconda，如未安装可参考：pytorch系列（二）Linux下安装anaconda（还没写，更新之后再加链接，小伙伴们先查看其他anaconda教程吧！）

1、创建新的pytorch虚拟环境

根据自己的python版本在终端输入如下命令，将新环境命名为：pytorch（你喜欢的任意名字均可）

conda create -n pytorch python=3.7

显示内容如下， proceed的那行输入：y

2、进入新创建的pytorch环境

source activate pytorch

activate后面的pytorch就是上面起的名字，按照你自己的输入即可。
未来想要切换你安装在conda中的环境时也使用这句source activate + 名字 即可。

可以看到环境有base切换到了新创建的pytorch。

3、安装pytorch

进入pytorch官网，根据你的cuda版本选择：

把Run this Command显示的命令复制粘贴到终端即可。我的：

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

出现proceed那句，输入：y 即可

4、检测是否安装成功

在终端输入python，进入后import torch，成功则表示环境安装成功。

import torch