725 B
725 B
安装PyTorch
人工智能与机器学习课程的深度学习部分代码采用PyTorch框架。以下为安装说明:
1. 官网
以下为官方网站,其中给出了安装方式。
https://pytorch.org/
###2. 选择安装方式
需注意,安装时需注意考虑:
- 操作系统(Windows、Linux或MacOs)
- CPU或GPU,后者需安装cuda(指定版本)
在官网上,选择适当安装方式后,将给出在线安装命令。例如,用conda安装cuda10.2版本的PyTorch:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
将其拷贝到命令行中执行即可。
3. 建议
因为PyTorch可能较大,可以下载whl文件安装。