1 下载 Python

注：这一部分已不需要，可以直接使用 conda 下载

MacOS 和 Windows 下载 Python 全流程见 全平台下载安装 Python 全流程指南：Windows & MacOS

2 Conda 下载

利用 Conda 更好的管理开发环境，全流程见 Conda 创建虚拟环境全流程

3 创建环境并激活

3.1 首先创建环境

例如我们创建一个环境名为 pytorch 的虚拟环境

1
conda create -n pytorch

3.2 激活环境

而后激活这个环境

1
conda activate pytorch

激活后会在命令后/终端前显示环境名

1
2
(pytorch) Username/@xx %   # macOS
(pytorch) PC C:\xx\xx>     # Windows

3.3 下载 python 和 pip

在终端中输入

1
conda install pip

注：需要先激活环境。此时 conda 会自动为该环境配置 python 和 pip，为之后使用 pip 下载 torch 作准备。

4 官网查询下载 Pytorch 命令

去往 Pytorch 官网，向下滑动找到【Install PyTorch】

4.1 macOS 安装指令

根据实际情况，选择对应配置，复制代码后在【终端】运行即可。

  • 一定要在【步骤三：创建环境并激活】后执行
  • 目前 Pytorch 官方已不提供 conda 下载方式，但 conda 的环境管理仍然有效，采用 pip 安装同样可行

4.2 Windows 安装指令

根据实际情况，选择对应配置，复制代码后在【PowerShell】运行即可。

4.2.1 安装在 CPU

选择 CPU 后复制命令

4.2.2 安装在 GPU

在命令行中输入

1
nvidia-smi

则会返回英伟达显卡的 CUDA 版本，根据版本号选择

5 删除环境

因为是虚拟环境，删除环境不会对其他配置造成影响，在终端或命令行中输入

1
conda remove -n 环境名 --all