30 lines
725 B
Markdown
30 lines
725 B
Markdown
|
# 安装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文件安装。
|