深度学习（Pytorch）环境配置

深度学习（Pytorch）环境配置(炼丹第一步)

---

收藏+关注，炼丹不迷路！！

安装过程

1. 安装Anaconda，用于管理深度学习环境。
2. 安装Pytorch。
3. 检验安装结果。

---

1 安装Anaconda

​ Anaconda是一款管理包和环境的免费软件，可以帮助我们更方便简洁的管理Pytorch环境。可前往官网下载，如图1。

图1 Anaconda下载官网

按照图2中顺序及设置依次安装。

图2 Anaconda安装步骤

​ 安装完成后即可在开始的菜单栏中找到相关图标，如图3(左)中所示。点击Anaconda Powershell Prompt，弹出的窗口中出现base表示安装成功。

图3 Anaconda安装结果验证

---

2. 安装Pytorch

​ 在刚才打开的Anaconda (Powershell) Prompt中输入conda create -n pytorch python=3.8 后按回车键，创建安装Pytorch环境，图4为相关命令的解释。

conda create -n pytorch python=3.8

图4 环境创建命令

​ 在出现 Proceed（[y]/n）? 时输入y（下同），之后等待安装完成，如图5。

图5 安装过程

​ 在窗口中继续输入conda activate pytorch后按回车键，以切换到我们安装的环境中, 如图6。

conda activate pytorch

图6 切换环境

​ 完成上步切换环境后，打开pytorch官网，复制图7中指令。

图7 官网指令

​ 在窗口中继续输入从pytorch官网复制的指令conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia 安装torch和torchvision等包，这一步需要自己复制（显卡和驱动不同，指令也不同），如图8。

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

图8 torch和torchvision安装

---

3. 检验安装结果

​ 继续输入python 启动解释器，输入import torch 导入torch包，再输入torch.cuda.is_available()，输出True表示安装成功，如图9。

python

import torch

torch.cuda.is_available()

图9 安装完成