本文记录 PyTorch 环境配置及安装，欢迎交流。

运行环境

Windows 10
Python 3.7.8
PyTorch 1.6.0（Stable）
CUDA 10.2
GPU：NVIDIA GeForce GTX 1050 Ti
PyCharm Community Edition 2020.1.3
Jupyter Notebook 6.0.1

配置 Anaconda

在安装 Pytorch 前，我们需要安装 Anaconda，直接去官网下载就行，网上也有很多教程。在很多情况下，我们所做的工作可能需要不同版本的软件，例如我们有时需要 PyTorch 0.4，做另一个工作可能需要 PyTorch 1.0，这时需要卸载老版本，否则会出现各种头疼的问题。

在这里插入图片描述

而 Anaconda 为我们提供了解决方案，允许我们进行环境管理。Anaconda 中有不同的环境可供我们安装不同版本的软件，只需要使用 conda 指令切换环境，就可以实现软件版本的管理了。

使用下面的语句创建一个名为 pytorch 的环境：

# -n后面加环境名称，后面再加要安装的包
conda create -n pytorch python==3.7

下面显示将要安装的包，如有提示输入 y 即可安装：

在这里插入图片描述

安装完毕后，提示使用 conda activate pytorch 激活 pytorch 环境，使用 conda deactivate 退出当前环境：

在这里插入图片描述

还可以查看 Anaconda 中有哪些虚拟环境：

conda info --env
# 或者使用
conda env list

如图：

扫描二维码关注公众号，回复： 11546108 查看本文章

在这里插入图片描述

安装 PyTorch（附换源）

使用命令 pip list 可以列出当前环境下所有的包，可以发现目前还没有 pytorch 包：

进入 PyTorch 官网，选择稳定版本。现在 PyTorch 已经支持 Windows 系统了（以前是不支持的）。

在这里插入图片描述

直接将红色框中的命令输入到 Anaconda Peompt 中：

在这里插入图片描述

如果这个过程很慢或经常中断下载，其不支持断点下载令人很头疼，不过每次下载一堆包时，成功下载的包不用重新下。不妨试一下清华源加速安装，运行以下命令：

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
# reference
# https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/