6、PyTorch安装与模型设计训练全攻略

PyTorch安装与模型设计训练全攻略

1. PyTorch安装

PyTorch的安装方式有两种，既可以使用官方发布的二进制文件进行安装，也可以从源代码进行构建。为避免与其他工具产生潜在冲突，建议使用包管理器（如Anaconda）。目前最新版本是v1.3.1，为利用其前沿功能，我们将使用1.3版本，当然你也可以选择其他版本。

• Ubuntu系统 ：强烈建议使用Anaconda安装，这样不会影响系统自带的默认Python环境。
• Windows系统 ：若安装出现严重问题，可删除现有Python安装并重新安装所需版本。

1.1 安装官方二进制文件

如今在官网安装PyTorch变得十分简单，步骤如下：
1. 访问 https://pytorch.org/get-started/locally/ 。
2. 选择要安装的版本。
3. 选择操作系统。
4. 确定安装方式，如Conda、pip等。
5. 选择目标Python版本。
6. 选择使用的CUDA版本或是否不使用GPU。
7. 从页面底部网格框中选择并复制命令，粘贴到终端或命令提示符中运行。

若想将创建的Python环境设为系统默认Python环境，需添加环境变量，如 C:\Users\John\Anaconda3\envs\torch