PyTorch环境的配置及安装_pytorch环境搭建

1、安装Anaconda
https://repo.anaconda.com/
安装相应的版本，我这里安装的是Anaconda3-5.2.0-Windows-x86_64.exe

2、显卡准备
深度学习离不开显卡（PyTorch支持英伟达显卡）
检查显卡驱动是否正常安装：正常显示表示，驱动正常安装。

3、管理环境

3.1、

3.2、初始环境（base） (初始的屋子)
下面我们创建另一个环境（pytorch）:

conda create -n pytorch python=3.6
1

3.3、激活环境(pytorch):

conda activate pytorch
1

3.4、查看这个环境（屋子）中有哪些包

pip list
1

很明显，上面的包中没有pytorch包，所以我们要安装pytorch

3.5.安装pytorch
3.5.1、进入pytorch官网：https://pytorch.org/
英伟达显卡，如下操作：

选好后，复制运行指令：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
1

3.5.2、查看驱动版本是否符合conda要求
命令窗口 输入nvidia-smi，查看驱动版本。
conda仅支持驱动版本号大于396.26

3.5.3,、
在环境(pytorch)中运行3.5.1复制的指令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
1

再查看一下安装的包有哪些

pip list
1

很明显，pytorch包成功安装

3.6、检查pytorch是否安装成功
在pytorch环境中 输入 python
输入import torch若没有报错，正常显示，则pytorch成功安装。

3.7、检查pytorch是否可用使用你电脑的GPU
输入torch.cuda.is_available()
如果显示True，则你的这台计算机的GPU可以给pytorch使用

至此，大功告成。

4.总结

• 创建conda环境，指定python版本
• 激活环境使用
• 安装pytorch