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