# 安装PyTorch

**人工智能与机器学习**课程的深度学习部分代码采用PyTorch框架。以下为安装说明：

### 1. 官网

以下为官方网站，其中给出了安装方式。

```
https://pytorch.org/
```

###2. 选择安装方式

需注意，安装时需注意考虑：

* 操作系统（Windows、Linux或MacOs）
* CPU或GPU，后者需安装cuda（指定版本）

在官网上，选择适当安装方式后，将给出在线安装命令。例如，用conda安装cuda10.2版本的PyTorch：

```shell
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
```

将其拷贝到命令行中执行即可。

### 3. 建议

因为PyTorch可能较大，可以下载whl文件安装。